POJ2831Can We Build This One?(次小生成树)

题意：给你一幅图，再给你Q个询问，每个询问为id cost，即如果将id这条边的边权改为cost的话，这条边是否可能是最小生成树中的一条边
思路：这道题主要是先用prim算法求出i,j结点之间的最大边的权值，用f[i][j]记录。然后后面查询的时候，查a,b两结点的边的权值如果小于f[a][b]那么就可以产生最小生成树，否则不行。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define inf 0x3f3f3f3f
using namespace std;
struct node
{
	int u,v,c;
}s[100009];
int mp[1005][1005],f[1005][1005],pre[1005];
int n,m;
void init()
{
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j<=n;j++)
		{
			if(j!=i)
				mp[j][i]=mp[i][j]=inf;
			else
				mp[i][i]=0;
		}
	}
}
void prim()
{
	int vis[1005],dis[1005];
	memset(vis,false,sizeof(vis));
	for(int i=1;i<=n;i++) dis[i]=mp[1][i],pre[i]=1;
	dis[1]=0;vis[1]=true;
	for(int i=1;i<n;i++)
	{
		int tmp=inf,k,src;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&dis[j]<tmp)
			{
				tmp=dis[j];
				k=j;
			}
		}
		vis[k]=true;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&dis[j]>mp[k][j])
			{
				dis[j]=mp[k][j];
				pre[j]=k;
			}
		}
		src=pre[k];
		f[k][src]=f[src][k]=tmp;
		for(int j=1;j<=n;j++)
		{
			if(vis[j]&&j!=src&&j!=k)
			{
				f[j][k]=max(f[j][src],tmp);
				f[k][j]=f[j][k];
			}
		}
	}
}
int main()
{
	//freopen("t.txt","r",stdin);
	int q,a,b,c;
	while(scanf("%d%d%d",&n,&m,&q)!=EOF)
	{
		init();
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			if(c<mp[a][b])
			mp[a][b]=mp[b][a]=c;
			s[i].u=a;s[i].v=b;s[i].c=c;
		}
		prim();
		/*for(int i=1;i<=n;i++)
			{
			for(int j=1;j<=n;j++)
			printf("%d ",f[i][j]);
			printf("\n");
			}*/
		while(q--)
		{
			scanf("%d%d",&a,&b);
			int u=s[a].u;
			int v=s[a].v;
			if(f[u][v]>=b)
				printf("Yes\n");
			else
				printf("No\n");
		}
	}
 }