Dijkstra算法

目录

写法一：如果有负边权的话会出错，时间复杂度为：O(n^2)

写法二：堆优化——用优先队列维护被更新的点的集合（也是不能有负的权值）时间复杂度为O(mlogm)  

注意：其实主要被背第二种即可

---

写法一：如果有负边权的话会出错，时间复杂度为：O(n^2)

例题：5.星际传送门网络 - 蓝桥云课 (lanqiao.cn)

#include<bits/stdc++.h>//万能头
#define int long long//直接将int定义为long long这样不用考虑是不是会超出范围，但是要注意int main要改成signed main
#define inf INT_MAX / 2//设置为无穷大
using namespace std;

const int N = 1e3 + 10;//定义最大的数据范围
int n , m , q;//一共有n个数，m次输入，q次询问
struct edge//定义结构体
{
  int v , w;//v表示终点，w表示权值
};
vector<edge> e[N];//用来存放各条边
int d[N] , vis[N];//d[N]表示原点到达i的距离，vis[N]表示这个点有没有删除

void dijkstra(int s, int o)//dijkstra算法模板
{
  for(int i = 0 ; i <= n ; i++) vis[i] = 0;//将之前的记录清零
  for(int i = 0 ; i <= n ; i++) d[i] = inf;//将一开始原点到所有点的距离设为无穷大
  d[s] = 0;//原点到原点的距离为0
  for(int i = 1 ; i < n ; i++)//循环n - 1次
  {
    int u = 0;//一个哨位，这个原点到这个位置永远是无穷大
    for(int j = 1 ; j <= n ; j++)//循环所有的点
    if(!vis[j] && d[j] < d[u]) u = j;//寻找没有被删除，且离选中的点距离最小的点
    vis[u] = 1;//找到这个点将他删除
    for(auto ed : e[u])//循环判断找到的这个点所有的邻边
    {
      int v = ed.v , w = ed.w;//将这个点的终点与权值表达出来
      if(d[v] > d[u] + w)//如果权值变小了，就更新为小的权值
      d[v] = d[u] + w;
    }
  }
  if(d[o] == inf) cout << -1 << endl;//判断是不是无穷大，如果无穷大就表示不能到达
  else cout << d[o] << endl;//否则输出到这个点的距离
}

signed main()//因为把int定义为long long所以不能用int要用signed
{
  cin >> n >> m >> q;
  for(int i = 0 ; i < m ; i++)
  {
    int a , b , c;//a表示起点，b表示终点，c表示权值
    cin >> a >> b >> c;
    e[a].push_back({b , c});//因为是无向图，所以两种情况可以相反，所以都要录入
    e[b].push_back({a , c});
  }
  while(q--)//q次询问
  {
    int u , v;//每次询问的起点以及终点
    cin >> u >> v;
    dijkstra(u , v);//算法
  }
  return 0;
}

写法二：堆优化——用优先队列维护被更新的点的集合（也是不能有负的权值）时间复杂度为O(mlogm)  

创建一个pair类型的大根堆q{-距离 ， 点}，把距离取负值，距离最小的元素最大，一定在堆顶。

1.初始化，{0 ， s}入队 ， d[s] = 0，d[其它点] = +无穷；
2.从队头淡出距离最小的点u，把u扩展过则跳过，否则打上标记；
3.对u的所有出边执行松弛操作，把{-d[v] , v}压入队尾；
4.重复2、3步操作，直到队列为空  

模板：

struct edge{int v , w;};//定义结构体
vector<edge> e[N];
int d[N] , vis[N];//d[N]表示第i个数到原点的距离，vis[N]表示是否被标记
int pre[N];//用来表示路径的，也就是记录每个点的前驱点
priority_queue<pair<int , int>> q;//优先队列

void dijkstra(int s)
{
  for(int i = 0 ; i <= n ; i++) d[i] = inf;//inf为最大值
    d[s] = 0; q.push({0 , s});
  while(q.size())
  {
    auto t = q.top() ; q.pop();
    int u = t.second;
    if(vis[u]) continue;//再出对跳过
    vis[u] = 1;//否则标记u已经出队
    for(auto ed : e[u])
    {
       int v = ed.v , w = ed.w;
       if(d[v] > d[u] + w)
       {
          d[v] = d[u] + w;
          pre[v] = u;//记录前驱点
          q.push({-d[v] , v});//大根堆
       }
    }
  }
}

void dfs_path(int u)//递归输出路径
{
  if(u == s) {cout << u << " "; return ;}//到达原点就结束
  dfs_path(pre[u]);
  cout << u << " " ;
}

注意：其实主要被背第二种即可