1129. 热浪单源最短路板子题

Posted 2021-10-30 幽殇默

https://www.acwing.com/problem/content/1131/
朴素版Dijkstra()

#include<bits/stdc++.h>
using namespace std;
const int N=2510;
int g[N][N],st[N],n,t,startx,endx;
int dist[N];
void Dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[startx]=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++) if(!st[j]&&(t==-1 || dist[j]<dist[t])) t=j;
        st[t]=1;
        for(int j=1;j<=n;j++) dist[j]=min(dist[j],dist[t]+g[t][j]);
    }
}
int main(void)
{
    memset(g,0x3f,sizeof g);
    cin>>n>>t>>startx>>endx;
    for(int i=0;i<t;i++)
    {
        int a,b,c; cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);
    }
    Dijkstra();
    cout<<dist[endx]<<endl;
    return 0;
}

堆优化版的Dijkstra()

#include<bits/stdc++.h>
using namespace std;
const int N=1e5*2+10;
typedef pair<int,int> PII;
int h[N],e[N],w[N],ne[N],idx;
int n,t,startx,endx;
int st[N],dist[N];
void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void Dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[startx]=0;
    priority_queue<PII,vector<PII>,greater<PII>> heap; heap.push({0,startx});
    while(heap.size())
    {
        auto t=heap.top(); heap.pop();
        int u=t.second,d=t.first;
        if(st[u]) continue;
        st[u]=1;
        for(int i=h[u];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>d+w[i])
            {
                dist[j]=d+w[i];
                heap.push({dist[j],j});
            }
        }
    }
}
int main(void)
{
    cin>>n>>t>>startx>>endx;
    memset(h,-1,sizeof h);
    while(t--)
    {
        int a,b,c; cin>>a>>b>>c;
        add(a,b,c),add(b,a,c);
    }
    Dijkstra();
    cout<<dist[endx];
    return 0;
}

spfa()

#include<bits/stdc++.h>
using namespace std;
const int N=1e5*2+10;
int h[N],e[N],w[N],ne[N],idx;
int n,t,startx,endx;
int dist[N],st[N];
void add(int a,int b,int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
void spfa()
{
    memset(dist,0x3f,sizeof dist);
    dist[startx]=0; st[startx]=1;
    queue<int>q; q.push(startx);
    while(q.size())
    {
        int u=q.front(); q.pop();
        st[u]=0;
        for(int i=h[u];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(dist[j]>dist[u]+w[i])
            {
                dist[j]=dist[u]+w[i];
                if(!st[j]) q.push(j),st[j];
            }
        }
    }
}
int main(void)
{
    memset(h,-1,sizeof h);
    cin>>n>>t>>startx>>endx;
    while(t--)
    {
        int a,b,c; cin>>a>>b>>c;
        add(a,b,c),add(b,a,c);
    }
    spfa();
    cout<<dist[endx];
    return 0;
}