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
*/