题意:给你一幅图,再给你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");
}
}
}