dijkstra模板

Posted 2021-01-24 lyhhahaha

传送门

不知道为什么这一题要用pair<int,int>才能过，感觉具体的实现过程还有几个细节没弄懂，暂且就这样吧等有时间再研究一下

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
int dis[100005],book[100005];
const int inf=2147483647;
int n,m,s;
struct p
{
    int to,w;
};
vector<p>g[100005];
typedef pair<int,int>pp;//第一位保存最短距离，第二位保存节点编号 
priority_queue<pp,vector<pp>,greater<pp> >q;//优先队列，优先返回较小值 （默认先排 pair 的第一位）
void dijkstra()
{
    int i;
    for(i=1;i<=n;i++)dis[i]=inf;
    dis[s]=0;
    q.push(pp(0,s));
    while(q.size()) 
    {
        pp x=q.top();
        q.pop();
        if(book[x.second]==1)continue;//这个点出堆被标记过了就continue
        book[x.second]=1;//标记出过一次堆 
        for(i=0;i<g[x.second].size();i++)
        {
            int y=g[x.second][i].to;//x是起点，y是终点 
            if(dis[y]>dis[x.second]+g[x.second][i].w)
            {
                dis[y]=dis[x.second]+g[x.second][i].w;
                q.push(pp(dis[y],y));
            }
        } 
    }
}
void solve()
{
    int i;
    cin>>n>>m>>s;
    for(i=1;i<=m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        g[a].push_back((p){b,c});
    }
    dijkstra();
    for(i=1;i<=n;i++)cout<<dis[i]<<" ";
}
int main()
{
    solve();
}