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

#include<bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
#define ll long long
using namespace std;
typedef pair<int,int> PII;
const int N=1010;
int g[N][N];//该题边数为n方，不适宜用kruskal算法
int dist[N];
bool vis[N];
int val[N];//存储每个结点的权值
int t,n;
void init()
{
    for(int i=0;i<=n;i++)
    {
        dist[i]=INF;
        vis[i]=false;
        for(int j=0;j<=n;j++)//初始化邻接矩阵
        {
            if(i==j)g[i][j]=0;
            else g[i][j]=3*(val[i]+val[j])+7;
        }
    }
}
int prim()
{
    int res=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&(t==-1||dist[j]<dist[t]))
                t=j;
        }
        vis[t]=true;
        if(i)res+=dist[t];
        for(int j=1;j<=n;j++)
        {
            if(!vis[j])dist[j]=min(dist[j],g[t][j]);
        }
    }
    return res;
}
void solve()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>val[i];
    }
    init();
    int res=prim();
    cout<<res<<endl;
}
int main()
{
    cin>>t;
    while(t--)
    {
        solve();
    }
    return 0;
}