算法競賽——進階指南——acwing342. 道路與航線 spfa寫法(slf優化)

spfa+雙端隊列的小優化,能卡過。。。。

以後遇事不決莽spfa再說。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 1e5+7;

int head[M],cnt;
void init(){cnt=0,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,val;}ee[M*2];
void add(int x,int y,int w){ee[++cnt].nxt=head[x],ee[cnt].val=w,ee[cnt].to=y,head[x]=cnt;}

int d[M],vs[M];
void spfa(int s)
{
	memset(d,60,sizeof(d));
	memset(vs,0,sizeof(vs));
	deque<int>q;
	d[s]=0;vs[s]=1;q.push_front(s);
	while(q.size())
	{
		int x=q.front();q.pop_front();
		vs[x]=0;
		for(int i=head[x];i;i=ee[i].nxt)
		{
			int y=ee[i].to,w=ee[i].val;
		//	cout<<x<<" "<<y<<endl;
			if(d[y]>d[x]+w)
			{
				d[y]=d[x]+w;
				if(!vs[y])
				{
					vs[y]=1;
				//	cout<<"okok"<<endl;
					if(q.size()&&d[y]<d[q.front()])//先用小的更新, 
						q.push_front(y);
					else 
						q.push_back(y);
				}
			}
		}
	}
	return ;
}

int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	int n,R,P,S,x,y,w;
  	cin>>n>>R>>P>>S;
  	for(int i=1;i<=R;i++)
  	{
  		cin>>x>>y>>w;
  		add(x,y,w);
  		add(y,x,w);
	}
	for(int i=1;i<=P;i++)
	{
		cin>>x>>y>>w;
		add(x,y,w);
	}
	spfa(S); 
	for(int i=1;i<=n;i++)
	{
		if(d[i]==d[n+1])cout<<"NO PATH"<<endl;
		else cout<<d[i]<<endl;
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章