天天写算法之Connect the Cities

训练的prime算法太少，都用Kruskal就过了，mark一下。

地址 点击打开链接

//546MS     1220K    1441B     C++    
//G++ TLE，prim算法模板题 
#include<stdio.h>
#include<string.h>
#define N 505
#define INF 0x7ffffff
int g[N][N];
int low[N],vis[N];
int n;
int prim()
{
    int pos,res=0;
    memset(vis,0,sizeof(vis));
    vis[1]=1;
    pos=1;
    for(int i=1;i<=n;i++)
        if(i!=pos) low[i]=g[pos][i];
    for(int i=1;i<n;i++){
        int minn=INF;
        pos=-1;
        for(int j=1;j<=n;j++)
            if(!vis[j] && minn>low[j]){
                minn=low[j];
                pos=j;
            }
        if(pos==-1) return -1; //注意这里 
        res+=minn;
        vis[pos]=1;
        //printf("**%d %d\n",pos,minn);
        for(int j=1;j<=n;j++)
            if(!vis[j] && low[j]>g[pos][j])
                low[j]=g[pos][j];
    }
    return res;
}
int main(void)
{
    int t,m,k;
    int u,v,d;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<=n;i++) 
            for(int j=0;j<=n;j++)
                g[i][j]=INF;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&u,&v,&d);
            if(g[u][v]>d)    //notice 
                g[u][v]=g[v][u]=d; 
        }
        for(int i=0;i<k;i++){
            scanf("%d",&d);
            scanf("%d",&u);
            d--;
            while(d--){
                scanf("%d",&v);
                g[u][v]=g[v][u]=0;
                u=v;
            }
        }
        printf("%d\n",prim());
    }
    return 0;
}