C - 小Y上学记——认识新同学

C - 小Y上学记——认识新同学

Time Limit: 4000/2000MS (Java/Others)    Memory Limit: 128000/64000KB (Java/Others)

Submit Status

Problem Description

小Y来到了一个新的班级，新的集体，第一件事肯定是要去认识新同学。

然而想认识全班同学，所需要的代价是很大的，所以小Y想让代价尽可能小。

认识新同学有两种办法，一种是直接去认识，另一种是通过已经认识的同学去认识新同学。

小Y想知道最小代价是多少呢？

Input

多组数据，每组数据首先是一个N，表示新的班级的同学数（包括小Y，2<=N<=1000）

接下来是N-1个整数，表示小Y直接认识第i名同学所需要的代价。

接下来是一个N-1阶矩阵，其中第i行第j列表示通过第i名同学认识第j名同学的代价。

其中保证对角线为0，且保证对称.

所有代价均为不超过1000的正整数。

Output

对于每组数据，输出一个整数，表示最少代价。

Sample Input

3
2 4
0 1
1 0
4
2 4 3
0 1 2
1 0 1
2 1 0

Sample Output

3
4

可以转化成最小生成树来求解即可：

代码：2015.7.30（自己写的还是比不上网上大神的模板，唉。。。）

#include <iostream>
#include <stdio.h>
#include <queue>
using namespace std;
#define MAX 11010
int ID[MAX];
int Sum;
struct Node
{
    int a,b,v;
    friend bool operator <(Node a,Node b)
    {
        return a.v>b.v;
    }
};
void Cread(int N)
{
    for(int i=1;i<=N;i++){ID[i]=i;}
}
int Find(int x)
{
    int TMD=x,TMP;
    while(TMD!=ID[TMD])TMD=ID[TMD];
    while(x!=TMD){TMP=ID[x];ID[x]=TMD;x=TMP;}
    return x;
}
int main()
{
    int N,i,j,a,b,A,B,Q,SIGN;
    Node NUM;
    while(scanf("%d",&N)!=EOF)
    {
        priority_queue<Node>ID_N;
        Cread(N);Sum=0;SIGN=0;
        for(i=2;i<=N;i++)
        {
            NUM.a=1;NUM.b=i;
            scanf("%d",&NUM.v);
            ID_N.push(NUM);
        }
        for(i=2;i<=N;i++)
        {
            for(j=2;j<=N;j++)
            {
                NUM.a=i;NUM.b=j;
                scanf("%d",&NUM.v);
                if(i==j)NUM.v=MAX;
                else ID_N.push(NUM);
            }
        }
        while(!ID_N.empty())
        {
            if(N==SIGN+1)break;
            NUM=ID_N.top();ID_N.pop();
            A=Find(NUM.a);B=Find(NUM.b);
            if(A!=B)
            {
                ID[A]=B;
                Sum+=NUM.v;
                SIGN++;
            }
        }
        printf("%d\n",Sum);
    }
    return 0;
}