POJ2831Can We Build This One?(次小生成樹)
阿新 • • 發佈:2018-12-11
題意:給你一幅圖,再給你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"); } } }