QS Network ZOJ - 1586 （最小生成树kruskal）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Miranda_ymz/article/details/86591493

QS Network ZOJ - 1586 （最小生成树）

题意：给你t表示数据，每组数据有：n表示点得数量。接下来一行有n个数表示每个点使用得适配器得价格。接下来n*n得矩阵表示每两个点之间的修建网线的价格。

思路：板子题把，使用kruskal的板子，直接将边权值加进去的时候顺便加入两个点的值。过一遍就好了。没什么坑

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
//最小生成树 
const int maxn=1000+10;//最大点数
const int maxm=maxn*maxn;
int a[maxn];
int f[maxn];//并查集使用
struct Edge{
	int u,v,w;
	bool operator <(const Edge &u)const{
		return w<u.w;
	}
}edge[maxm];
int tot;
void addedge(int u,int v,int w)
{
	edge[tot].u=u;
	edge[tot].v=v;
	edge[tot++].w=w;
} 
int find(int x)
{
	if(f[x]==-1) return x;
	return f[x]=find(f[x]);
}
int kruskal(int n)
{
	memset(f,-1,sizeof(f));
	sort(edge,edge+tot);
	int cnt=0;
	int ans=0;
	for(int i=0;i<tot;i++)
	{
		int u=edge[i].u;
		int v=edge[i].v;
		int w=edge[i].w;
		int t1=find(u);
		int t2=find(v);
		if(t1!=t2)
		{
			ans+=w;
			f[t1]=t2;
			cnt++;
		}
		if(cnt==n-1) break;
	}
	if(cnt<n-1) return -1;
	else return ans; 
}
int main() 
{
	int t,n,w;
	scanf("%d",&t);
	while(t--)
	{
		tot=0;
		scanf("%d",&n);
		for(int i=0;i<n;i++)
		scanf("%d",&a[i]);
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			{
				scanf("%d",&w);
				if(i==j) continue;
				addedge(i,j,w+a[i]+a[j]);
			}
		}
		//cout<<tot<<endl; 
		int ans=kruskal(n);
		printf("%d\n",ans);
	}
	return 0;
}