【JZOJ3237】间谍派遣【最小生成树】

题目大意：

题目链接：https://jzoj.net/senior/#main/show/3237
给出$n$个点，连接任意两个点$i,j$需要花费$a[i,j]$，将任意点$i$连向源点需要花费$b[i]$，求最小花费。

---

思路：

这又是一道裸题。。。
题目描述说的多么高大上啊什么，害得我一开始敲了一个费用流，结果发现就是最小生成树。。。

诶最小生成树的思路就不用我说了吧。

---

代码：

#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;

const int N=1210;
int n,S,father[N*N];
ll ans;

struct node
{
	int x,y,cost;
}a[N*N];

int C(int x,int y)
{
	return (x-1)*n+y;
}

bool cmp(node x,node y)
{
	return x.cost<y.cost;
}

int find(int x)
{
	return x==father[x]?x:father[x]=find(father[x]);
}

int main()
{
	scanf("%d",&n);
	S=N-1;
	for (int i=1;i<=n;i++)
		for (int j=1;j<=n;j++)
		{
			int id=C(i,j);
			scanf("%d",&a[id].cost);
			a[id].x=i; a[id].y=j;
		}
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[n*n+i].cost);
		a[n*n+i].x=S; a[n*n+i].y=i;
	}
	sort(a+1,a+1+n*n+n,cmp);
	for (int i=1;i<=n*n+n;i++)
		father[i]=i;
	for (int i=1;i<=n*n+n;i++)
	{
		int x=find(a[i].x),y=find(a[i].y);
		if (x!=y)
		{
			ans+=(ll)a[i].cost;
			father[x]=y;
		}
	}
	printf("%lld\n",ans);
	return 0;
}