1. 程式人生 > >POJ - 4046 Sightseeing 暴力列舉下的SPFA

POJ - 4046 Sightseeing 暴力列舉下的SPFA

題目連結:點選檢視

題意:給一個圖和q個詢問,每個詢問查詢圖中兩點的(距離+路徑上最大值)的最小值。

題解:暴力列舉每個點,以這個點為起點並且為路徑上所有經過的點的價值中最大的點來跑最短路,每跑一次,更新下q次詢問

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f3f3f3f3f
const int N=1100;
struct node{
	int to,nex;
	ll d;
}e[20010*2];
int head[N],vis[N],val[N],len;
ll dis[N],ans[20010];
int st[20010],en[20010];
int n,m,qq;
void init()
{
	memset(head,-1,sizeof(head));
	len=0;
}
void add(int x,int y,ll z)
{
	e[len].to=y;
	e[len].d=z;
	e[len].nex=head[x];
	head[x]=len++;
}
void SPFA(int ss)
{
	for(int i=1;i<=n;i++)
	{
		dis[i]=INF;
		vis[i]=0;
	}
	queue<int> q;
	q.push(ss);
	dis[ss]=0;vis[ss]=1;
	while(!q.empty())
	{
		int now=q.front();q.pop();
		vis[now]=0;
		for(int i=head[now];~i;i=e[i].nex)
		{
			int to=e[i].to;
			if(dis[to]>dis[now]+e[i].d&&val[ss]>=val[to])
			{
				dis[to]=dis[now]+e[i].d;
				if(!vis[to])
				{
					vis[to]=1;
					q.push(to);	
				}
			}
		}
	}
	for(int i=1;i<=qq;i++)
	{
		if(dis[st[i]]!=INF&&dis[en[i]]!=INF&&ans[i]>dis[st[i]]+dis[en[i]]+val[ss])
			ans[i]=dis[st[i]]+dis[en[i]]+val[ss];
	}
}
int main()
{
	int x,y;
	ll z;
	while(~scanf("%d%d",&n,&m)&&(n||m))
	{
		init();
		for(int i=1;i<=n;i++) scanf("%d",&val[i]);
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%lld",&x,&y,&z);
			add(x,y,z);
			add(y,x,z);
		}
		scanf("%d",&qq);
		for(int i=1;i<=qq;i++)
			scanf("%d%d",&st[i],&en[i]),ans[i]=INF;
		for(int i=1;i<=n;i++) SPFA(i);
		for(int i=1;i<=qq;i++)
		{
			if(ans[i]==INF) printf("-1\n");
			else printf("%lld\n",ans[i]);
		}		
		printf("\n");
	}
	return 0;
}
/*
6 7
1 2 3 4 5 6
1 2 1
2 3 2
3 4 3
4 5 4
1 5 5
2 5 2
1 4 3
5
1 4
2 3
1 5
3 5
1 6
2 1
10 20
1 2 5
1
1 2
0 0
*/