Light OJ 1040 Donation

  题意简单。

  思路嘛，就最小生成树，有几个坑，主要是重边的问题，数据比较小，乱搞一下就行了！开始没初始化，WA了一发！我太傻比！

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <map>
#include <vector>
#include <set>
#include <string>
#define PB push_back
#define FT first
#define SD second
#define MP make_pair
#define INF (0x3f3f3f3f)*2
using namespace std;
typedef long long LL;
typedef pair<int,int>  P;
const int maxn=55,MOD=7+1e9;
int G[maxn][maxn];
int N,dis[maxn],res;
bool vis[maxn];
bool  MST()  
{  
    memset(vis,0,sizeof(vis));
    res=0;
    dis[1]=0;  
    for(int i=2;i<=N;i++)  
        dis[i]=INF;  
    for(int i=2;i<=N;i++){  
        dis[i]=min(G[i][1],dis[i]); 
        dis[i]=min(G[1][i],dis[i]); 
    } 
    vis[1]=1;  
    for(int _=0;_<N;_++){  
        int _min=INF,u=-1;  
        for(int i=1;i<=N;i++){  
            if(!vis[i]&&_min>dis[i]){  
                _min=dis[i];  
                u=i;  
            }  
        }  
        if(u==-1) break;  
        vis[u]=1;  
        res+=_min;
        for(int i=1;i<=N;i++){ 
            if(!vis[i]&&dis[i]>G[u][i]) dis[i]=G[u][i];
            if(!vis[i]&&dis[i]>G[i][u]) dis[i]=G[i][u];

        }  
    }  
    for(int i=1;i<=N;i++)  
        if(dis[i]==INF) return 0;  
    return 1;  
}  
int main()
{
        //freopen("E:/ACM_code/in.txt","r",stdin);
        //freopen("out.txt","w",stdout);
        int T,ca=0;
        scanf("%d",&T);
        while(T--){
            scanf("%d",&N);
            int sum=0;
            for(int i=1;i<=N;i++)
                for(int j=1;j<=N;j++){
                    scanf("%d",&G[i][j]);
                    sum+=G[i][j];
                    if(G[i][j]==0||i==j) G[i][j]=INF;
                }
            //printf("%d\n",sum); 
            if(MST()) printf("Case %d: %d\n",++ca,sum-res);
            else printf("Case %d: -1\n",++ca);
        }
        //system("pause");
        return 0;
}