poj2831生成树

链接：http://poj.org/problem?id=2831

题意：给出一些边，再对这些边进行修改（只允许边权值减小），问该边修改之后是否可能是最小生成树上的边。

思路：用prime算法求最小生成树，边求边计算出从s到t之间最长的边。然后询问时候，看这条边和s到t之间的最长的边比较大小就可以了~

一开始想的很麻烦~

 #include <stdio.h>
 #include <string.h>
 const int N=1005;
 const int inf=100000000;
 int map[N][N],n;
 int mx[N][N];
 int vis[N];
 int pre[N];
 int dis[N];
 struct node{
    int s,t;
 }edge[N*100];
 int max(int a,int b){return a>b?a:b;}
 int min(int a,int b){return a<b?a:b;}
 void Prime(){
     int i,j,t,loc;
     memset(vis,0,sizeof(vis));
     memset(mx,0,sizeof(mx));
     for(i=1;i<=n;i++){
        dis[i]=inf;
     }
     dis[1]=0;loc=1;vis[1]=1;
     for(i=1;i<n;i++){
        for(j=1;j<=n;j++){
            if(!vis[j]&&map[loc][j]<dis[j]){
                dis[j]=map[loc][j],pre[j]=loc;
            }
        }
        int min=inf;
        for(j=1;j<=n;j++){
            if(!vis[j]&&dis[j]<min) min=dis[loc=j];
        }
        for(j=1;j<=n;j++) if(vis[j]){
            mx[loc][j]=max(mx[j][pre[loc]],dis[loc]);
            mx[j][loc]=mx[loc][j];
        }
        vis[loc]=1;
     }
 }
 int main(){
     int m,q,a,b,c;
     while(scanf("%d%d%d",&n,&m,&q)!=EOF){
         for(int i=1;i<=n;i++){
             for(int j=1;j<=n;j++){
                 map[i][j]=inf;
             }
         }
         for(int i=1;i<=n;i++) map[i][i]=0;

         for(int i=1;i<=m;i++){
             scanf("%d%d%d",&a,&b,&c);
             edge[i].s=a;edge[i].t=b;
             map[a][b]=map[b][a]=min(c,map[a][b]);
         }
         Prime();
         while(q--){
            scanf("%d%d",&a,&b);
            printf("%s\n",mx[edge[a].s][edge[a].t]>=b?"Yes":"No");
         }
     }
     return 0;
 }