次小生成树（poj2831）

来源：互联网发布：三凌plc编程软件下载编辑：程序博客网时间：2024/05/21 11:19

题目大意：给你一幅图，接下来是q个询问。每个询问的形式为：
id c
表示询问把第id条边的长度改为c后，这条边是否存在于整个图的最小生成树中。

做法：次小生成树。若用prim变形，记录一个数组mx[u][v] 表示在最小生成树中，连接u，v的路径中最长的一条边的长度。比较第id条边连接的u，v的mx[u][v] 。若c≤mx[u][v] ，则答案为Yes，否则为No。

#include<cstdio>#include<algorithm>#include<cstring>#include<cmath>using namespace std;const int maxn=1010;const int INF=0x3f3f3f3f;int g[maxn][maxn],mx[maxn][maxn];int x[maxn*100],y[maxn*100];int vis[maxn],pre[maxn],dis[maxn];int n,m,k;int e;void prim(){    memset(vis,0,sizeof(vis));    int i,j,k=1;    for(i=2;i<=n+1;i++) dis[i]=INF;    int _min;    for(i=1;i<=n;i++){        _min=INF;        for(j=1;j<=n;j++){            if(!vis[j] && dis[j]<_min){                k=j;                _min=dis[j];            }        }        for(j=1;j<=n;j++){            if(vis[j]){                mx[k][j]=max(mx[j][pre[k]],dis[k]);                mx[j][k]=mx[k][j];            }        }        vis[k]=1;        for(j=1;j<=n;j++){            if(!vis[j] && dis[j]>g[k][j]){                dis[j]=g[k][j];                pre[j]=k;            }        }    }}int main(){    scanf("%d%d%d",&n,&m,&k);    memset(g,INF,sizeof(g));    e=0;    int i,j;    for(i=1;i<=n;i++) g[i][i]=0;    for(i=1;i<=m;i++){        int w;        scanf("%d%d%d",&x[i],&y[i],&w);        g[x[i]][y[i]]=g[y[i]][x[i]]=min(g[x[i]][y[i]],w);    }    prim();    while(k--){        int a,b;        scanf("%d%d",&a,&b);        if(mx[x[a]][y[a]]<b) printf("No\n");        else printf("Yes\n");    }    return 0;}

^_^

1 0