DIJ - Heat Wave G(热浪) - 洛谷 P1339

最新推荐文章于 2022-04-22 22:50:00 发布

njuptACMcxk

最新推荐文章于 2022-04-22 22:50:00 发布

阅读量306

点赞数

分类专栏：最短路文章标签：算法图论数据结构 dijkstra acm竞赛

本文链接：https://blog.csdn.net/njuptACMcxk/article/details/106951645

版权

最短路专栏收录该内容

48 篇文章 0 订阅

订阅专栏

DIJ - Heat Wave G - 洛谷 P1339

题目描述
有一个 n 个点 m 条边的无向图，请求出从 s 到 t 的最短路长度。

输入格式
第一行四个正整数 n,m,s,t。接下来 m 行，每行三个正整数 u,v,w表示一条连接 u,v 长为 w 的边。

输出格式
输出一行一个整数，表示答案。

输入输出样例
输入

输出

说明/提示
【数据范围】
对于 100% 的数据，1≤n≤2500，1≤m≤6200，1≤w≤1000。

【样例说明】
5→6→1→4 为最短路，长度为 3+1+3=7。

$模板题，由于 1 \leq n \leq 2500 ， 1 \leq m \leq 6200 ，$

$①、朴素版dijkstra算法的时间复杂度为O(n^2)，可行。$

$② 、堆优化 d i j k s t r a 算法的时间复杂度为 O (m l o g n) ，可行。$

$③ 、 s p f a 算法的时间复杂度为 O (n m) ，可行。$

$这里注意无向图边的数量要开两倍。$

$这里贴出堆优化 d i j k s t r a 代码：$

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>

#define P pair<int,int>

using namespace std;

const int N=2510,M=6200*2+10;

int n,m,S,T,e[M],ne[M],h[N],w[M],idx;
int dis[N];
bool st[N];

void add(int a,int b,int c)
{
    e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;    
}

int dijkstra()
{
    memset(dis,0x3f,sizeof dis);
    dis[S]=0;
    
    priority_queue<P,vector<P>,greater<P> > heap;
    heap.push({0,S});
    
    while(heap.size())
    {
        P t=heap.top();
        heap.pop();
        
        int id=t.second;
        if(st[id]) continue;
        st[id]=true;
        
        for(int i=h[id];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]>dis[id]+w[i])
            {
                dis[j]=dis[id]+w[i];
                heap.push({dis[j],j});
            }
        }
    }
    
    return dis[T];
}

int main()
{
    memset(h,-1,sizeof h);
    
    cin>>n>>m>>S>>T;
    int a,b,c;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    
    cout<<dijkstra()<<endl;
    
    return 0;
}