次小生成树（poj2831）

题目大意：给你一幅图，接下来是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;
}

^_^