【图论 单源最短路】100276. 最短路径中的边

简介: 【图论 单源最短路】100276. 最短路径中的边

本文时间知识点

单源最短路

图论知识汇总

LeetCode100276. 最短路径中的边

给你一个 n 个节点的无向带权图,节点编号为 0 到 n - 1 。图中总共有 m 条边,用二维数组 edges 表示,其中 edges[i] = [ai, bi, wi] 表示节点 ai 和 bi 之间有一条边权为 wi 的边。

对于节点 0 为出发点,节点 n - 1 为结束点的所有最短路,你需要返回一个长度为 m 的 boolean 数组 answer ,如果 edges[i] 至少 在其中一条最短路上,那么 answer[i] 为 true ,否则 answer[i] 为 false 。

请你返回数组 answer 。

注意,图可能不连通。

示例 1:

输入:n = 6, edges = [[0,1,4],[0,2,1],[1,3,2],[1,4,3],[1,5,1],[2,3,1],[3,5,3],[4,5,2]]

输出:[true,true,true,false,true,true,true,false]

解释:

以下为节点 0 出发到达节点 5 的 所有 最短路:

路径 0 -> 1 -> 5 :边权和为 4 + 1 = 5 。

路径 0 -> 2 -> 3 -> 5 :边权和为 1 + 1 + 3 = 5 。

路径 0 -> 2 -> 3 -> 1 -> 5 :边权和为 1 + 1 + 2 + 1 = 5 。

示例 2:

输入:n = 4, edges = [[2,0,1],[0,1,1],[0,3,4],[3,2,2]]

输出:[true,false,false,true]

解释:

只有一条从节点 0 出发到达节点 3 的最短路 0 -> 2 -> 3 ,边权和为 1 + 2 = 3 。

提示:

2 <= n <= 5 * 104

m == edges.length

1 <= m <= min(5 * 104, n * (n - 1) / 2)

0 <= ai, bi < n

ai != bi

1 <= wi <= 105

图中没有重边。

单源最短路

准备修改 堆优化迪氏定理。优化过程中,发现直接使用 迪氏定理就可以了。浪费了20分钟。如果边{n1,n2,w} 是最短路径的边,则一定是以下情况之一:

image.png

计算这两条路径的最短路,就可以了。

提前计算好:0到各点的最短路径,(n-1)到各点的最短路。

时间复杂度: O(mlogm)+O(m) m是边数。

代码

class CNeiBo
{
public: 
  static vector<vector<int>> Two(int n, vector<vector<int>>& edges, bool bDirect, int iBase = 0) 
  {
    vector<vector<int>>  vNeiBo(n);
    for (const auto& v : edges)
    {
      vNeiBo[v[0] - iBase].emplace_back(v[1] - iBase);
      if (!bDirect)
      {
        vNeiBo[v[1] - iBase].emplace_back(v[0] - iBase);
      }
    }
    return vNeiBo;
  } 
  static vector<vector<std::pair<int, int>>> Three(int n, vector<vector<int>>& edges, bool bDirect, int iBase = 0)
  {
    vector<vector<std::pair<int, int>>> vNeiBo(n);
    for (const auto& v : edges)
    {
      vNeiBo[v[0] - iBase].emplace_back(v[1] - iBase, v[2]);
      if (!bDirect)
      {
        vNeiBo[v[1] - iBase].emplace_back(v[0] - iBase, v[2]);
      }
    }
    return vNeiBo;
  }
  static vector<vector<int>> Grid(int rCount, int cCount, std::function<bool(int, int)> funVilidCur, std::function<bool(int, int)> funVilidNext)
  {
    vector<vector<int>> vNeiBo(rCount * cCount);
    auto Move = [&](int preR, int preC, int r, int c)
    {
      if ((r < 0) || (r >= rCount))
      {
        return;
      }
      if ((c < 0) || (c >= cCount))
      {
        return;
      }
      if (funVilidCur(preR, preC) && funVilidNext(r, c))
      {
        vNeiBo[cCount * preR + preC].emplace_back(r * cCount + c);
      }
    };
    for (int r = 0; r < rCount; r++)
    {
      for (int c = 0; c < cCount; c++)
      {
        Move(r, c, r + 1, c);
        Move(r, c, r - 1, c);
        Move(r, c, r, c + 1);
        Move(r, c, r, c - 1);
      }
    }
    return vNeiBo;
  }
  static vector<vector<int>> Mat(vector<vector<int>>& neiBoMat)
  {
    vector<vector<int>> neiBo(neiBoMat.size());
    for (int i = 0; i < neiBoMat.size(); i++)
    {
      for (int j = i + 1; j < neiBoMat.size(); j++)
      {
        if (neiBoMat[i][j])
        {
          neiBo[i].emplace_back(j);
          neiBo[j].emplace_back(i);
        }
      }
    }
    return neiBo;
  }
};
typedef pair<long long, int> PAIRLLI;
class  CHeapDis
{
public:
  CHeapDis(int n,long long llEmpty = LLONG_MAX/10):m_llEmpty(llEmpty)
  {
    m_vDis.assign(n, m_llEmpty);
  }
  void Cal(int start, const vector<vector<pair<int, int>>>& vNeiB)
  {
    std::priority_queue<PAIRLLI, vector<PAIRLLI>, greater<PAIRLLI>> minHeap;
    minHeap.emplace(0, start);
    while (minHeap.size())
    {
      const long long llDist = minHeap.top().first;
      const int iCur = minHeap.top().second;
      minHeap.pop();
      if (m_llEmpty != m_vDis[iCur])
      {
        continue;
      }
      m_vDis[iCur] = llDist;
      for (const auto& it : vNeiB[iCur])
      {
        minHeap.emplace(llDist + it.second, it.first);
      }
    }
  }
  vector<long long> m_vDis;
  const long long m_llEmpty;
};
class Solution {
public:
  vector<bool> findAnswer(int n, vector<vector<int>>& edges) {
    auto neiBo = CNeiBo::Three(n, edges, false);
    CHeapDis dis1(n),dis2(n);
    dis1.Cal(0, neiBo);
    dis2.Cal(n - 1, neiBo);
    const long long llMinDis = dis1.m_vDis[n - 1];
    vector<bool> ret;
    for (const auto& v : edges) {
      bool b1 =  ( llMinDis ==(v[2] + dis1.m_vDis[v[0]] + dis2.m_vDis[v[1]]));
      bool b2 = (llMinDis == (v[2] + dis2.m_vDis[v[0]] + dis1.m_vDis[v[1]]));
      ret.emplace_back(b1 || b2);
    }
    return ret;
  }
};

扩展阅读

视频课程

有效学习:明确的目标 及时的反馈 拉伸区(难度合适),可以先学简单的课程,请移步CSDN学院,听白银讲师(也就是鄙人)的讲解。

https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了,为老板分忧,请学习C#入职培训、C++入职培训等课程

https://edu.csdn.net/lecturer/6176

相关下载

想高屋建瓴的学习算法,请下载《喜缺全书算法册》doc版

https://download.csdn.net/download/he_zhidan/88348653

我想对大家说的话
闻缺陷则喜是一个美好的愿望,早发现问题,早修改问题,给老板节约钱。
子墨子言之:事无终始,无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙,那算法就是他的是睛

测试环境

操作系统:win7 开发环境: VS2019 C++17

或者 操作系统:win10 开发环境: VS2022 C++17

如无特殊说明,本算法用**C++**实现。

相关文章
|
算法 C语言
最短路径——Dijkstra算法与Floyd算法
最短路径——Dijkstra算法与Floyd算法
218 0
最短路径——Dijkstra算法与Floyd算法
|
算法
最短路径算法
图的遍历算法(深度优先遍历,广度优先遍历)我们在之前文章已经讲过,今天我们一起来看一下最短路径算法。日常工作、学习常用的算法大部分已经更新到算法专栏中,欢迎大家关注我的算法专栏。
276 0
|
机器学习/深度学习 算法
迪杰斯特拉 (Dijkstra)算法求最短路径问题
迪杰斯特拉( Dijkstra )算法是典型最短路径算法,用于计算一个结点到其他结点的最短路径。它的主要特点是以起始点为中心向外层层扩展(广度优先搜索思想),直到扩展到终点为止。
150 0
最短路径算法——dijkstra
最短路径算法——dijkstra
最短路径算法——dijkstra
|
算法 固态存储
最短路径问题(迪杰斯特拉)算法
最短路径问题(迪杰斯特拉)算法
最短路径问题(迪杰斯特拉)算法
|
存储 算法
聊一聊利用Dijkstra求有向图的最短路径
我们都知道求最短路径有很多方法,比如Dijkstra算法、Bellman-Ford算法、Floyd-Warshall算法等等,这些算法各有优缺点,其中Floyd-Warshall算法时间复杂度较高,但是编码复杂度较小,而Bellman-Ford算法适用于处理有负权边的情况。至于本文要讲的Dijkstra算法,优点就是时间复杂度较小,但是不能处理有负权边的图。 我们需要根据不同情况选择不同的算法。
773 0
|
算法
图论——最短路——Dijkstra算法
对图论有一定了解的人,一定知道最短路。 最短路算法一共有4中,严格来说是3种,应为最后一个是第3个的优化。 他们分别是: Floyd、Dijkstra、Bellman-Ford和SPFA算法 Floyd是最暴力的思想,这里就不在阐述。
1599 0
|
算法 C++ BI
图论——强连通分量:Tarjan算法——练习1
上一次我们详细介绍了强连通分量的Tarjan算法,今天呢,我们来做一些习题来巩固Tarjan算法,毕竟它十分重要。 Tarjan算法详解 上面是上一次的详解,在做题时可供参考。 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~                               练习一般采用洛谷题库。
1319 0
|
算法
图论——强连通分量:Tarjan算法。
在有向图G中,如果两个定点u,v间存在一条u到v的路径,也存在一条v到u的路径,则称u,v是强连通的。 若有向图G的任意两点都强联通,则称G是一个强联通图。 非强连通图的极大强连通子图称为强连通分量。   这里,极大强连通子图可以理解为一个子图是强连通图,且它的任意子图都不是强联通。
2663 0