hdu 3371 Connect the cities （prim）

题目：http://acm.hdu.edu.cn/showproblem.php?pid=3371

一个普通的生成树，已连好的都把权值改为0；

但有竟然还有重边，这就没意思了。

#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=505;
const int maxcost=99999999;
int  graph[maxn][maxn],lowcost[maxn];
int prim(int n){  
    int lowcost[maxn];  
    int mst[maxn];  
    int i,j,minn,minid,sum=0;  
    for(i=2;i<=n;i++){             //lowcost初始化  
        lowcost[i]=graph[1][i];  
        mst[i]=1;  
    }  
    mst[1]=0;  
    for(i=2;i<=n;i++){  
        minn=maxcost;  
        minid=0;  
        for(j=2;j<=n;j++){     
            if(lowcost[j]<minn&&lowcost[j]!=-1){   //上一层已经把找过的赋为-1  
                minn=lowcost[j];  
                minid=j;  
            }  
        }  
        if(minn==maxcost)  
            return -1;        //不连通  
        //printf("%c - %c : %d\n", mst[minid] + 'A' - 1, minid + 'A' - 1, min);  
        sum+=minn;  
        lowcost[minid]=-1;  
        for(j=2;j<=n;j++){    //找新起点，谁小换谁,相同不换  
            if(graph[minid][j]<lowcost[j]){  
                lowcost[j]=graph[minid][j];  
                mst[j]=minid;   //mst中每一个都是minid，到下次时可以表示起点，然后 - 下一个minid  
            }  
        }  
    }  
    return sum;  
}  


int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        fill(graph[0],graph[0]+maxn*maxn,maxcost);
        fill(lowcost,lowcost+maxn,0);
        int n,m,k,a,b,val;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&val);
            if(graph[a][b]>val) graph[a][b]=graph[b][a]=val;
        }
        int con[maxn];
        for(int i=0;i<k;i++){
            int tt;
            scanf("%d",&tt);
            for(int j=0;j<tt;j++)
                scanf("%d",&con[j]);
            for(int v=0;v<tt;v++)
                for(int u=v+1;u<tt;u++)
                    graph[con[v]][con[u]]=graph[con[u]][con[v]]=0;
        }
        printf("%d\n",prim(n));
    }
    return 0;
}