1. 程式人生 > >POJ2831Can We Build This One?(次小生成樹)

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");
		}
	}
 }