SDUT2937->人活着系列之寻找最完美的人生

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
int Map[1001][1001];     //Map为初始计算的花费值
int dis[1111],vis[1111]; //Prim 算法有一个记录长度的dis数组和一个标记数组
void Prim(int n)
{
    memset(vis,0,sizeof(vis));   //初始化，由于多组输入呀
    int sum=0;
    vis[0]=1;                    //首先将
    int pos;                      //一个位置标记，标记在离出发点最近的且没有被标记过的点上，
    for(int i=0;i<n;i++)
        dis[i]=Map[0][i];           //初始化dis数组，假设从0开始，从0出发的所有边都放在dis里
    for(int i=1;i<n;i++)           //注意这里是n-1次！！！
    {
        int min_=INF;              //min_找离出发点最小的且没有被标记过的，并用pos标记，找完后再将vis数组标记，否则不对
        for(int j=0;j<n;j++)
        {
            if(dis[j]<min_&&vis[j]==0)
            {
                min_=dis[j];
                pos=j;
            }
        }
        vis[pos]=1;
        sum=sum+min_;             //每一轮找到个最小值，加到sum，最后sum即为答案
        for(int k=0;k<n;k++)     //更新dis数组
        {
            if(vis[k]==0&&Map[pos][k]<dis[k])
                dis[k]=Map[pos][k];
        }
    }
    cout<<sum<<endl;
}
int main()
{
    int t,n,x[1111];
    cin>>t;
    while(t--)
    {
        cin>>n;
        for(int i=0;i<n;i++)
            cin>>x[i];
        for(int i=0;i<n;i++)
        {
            for(int j=i+1;j<n;j++)       //这里简化了代码，只需看单向的，然后再双向存储就行了
            {
                int data=3*(x[i]+x[j])+7;
                Map[i][j]=data;
                Map[j][i]=data;
            }
        }
        Prim(n);
    }
    return 0;
}
/***************************************************
User name: ACM18171信科1801张林
Result: Accepted
Take time: 532ms
Take Memory: 2088KB
Submit time: 2019-01-12 00:43:48
****************************************************/