十分,dij+堆优化,求助
查看原帖
十分,dij+堆优化,求助
476919
爱国者楼主2021/7/27 15:01
#include<bits/stdc++.h>
#include<vector>
#define maxd 9223372036854775807
using namespace std;
priority_queue<int,vector<int>, greater<int> > f;
int n,m,s,x,y,w,head[1000010],cnt=0;
long long dist[10010];
bool us[100010],q[100010];
struct node
{
    int next,to,w;
}edge[100010];
void add(int u,int v,int w)
{
    edge[++cnt].to=v;
    edge[cnt].w=w;
    edge[cnt].next=head[u];
    head[u]=cnt;
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<=m;i++)
	{
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);
    }
    for(int i=1;i<=n;i++)
    	dist[i]=maxd;
    dist[s]=0; 
	f.push(s);
    while(f.empty()==0)
    {
    	int u=f.top();
    	f.pop();
    	us[u]=1;
    	for(int i=head[u];i;i=edge[i].next)
    	{
			if(dist[edge[i].to]>dist[u]+edge[i].w)
			{
				dist[edge[i].to]=dist[u]+edge[i].w;
				if((!us[edge[i].to])&&(!q[edge[i].to]))
				{
					f.push(edge[i].to);
					q[edge[i].to]=1;
				}
				
			}
		}
	}
    for(int i=1;i<=n;i++)
    {
		if(dist[i]==maxd)
			dist[i]=2147483647;
		printf("%lld ",dist[i]);
	}	
    return 0;

}
2021/7/27 15:01
加载中...