P1550 题解

题目大意

给 $n(n\le 300)$ 个点，每个点初始有一个值 $w_i$。任意两点间有一条边，记为 $p_{i,j}$。特殊的，$p_{i,j}=p_{j,i}$ 且 $p_{i,i}=0$。你要选点 $x$ 为根寻找最小生成树，并使得树的边权和与 $w_x$ 和最小。输出这个值。

样例输入

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

样例输出

9

思路

基础的最小生成树很简单，关键在于这题要选点作为根。我们考虑构造一个不存在的根（$0$），并将 $w_i$ 看作第 $p_{0,i}$ 并跑一个最小生成树即可。

代码

#include<bits/stdc++.h>
using namespace std;
int head[305],w[305],p[305][305],ans = 0,cnt = 0,n;
int find(int x) {
	return head[x] == x?x:head[x] = find(head[x]);
}
struct edge{
	int fr,to,we;
	bool operator <(const edge &x)const{
		return we < x.we;
	} 
}e[100005];
int main() {
	scanf("%d",&n);
	for(int i = 1;i <= n;i++) {
		scanf("%d",&w[i]);
		e[++cnt].fr = 0;
		e[cnt].to = i;
		e[cnt].we = w[i];
	}
	for(int i = 0;i <= n;i++) head[i] = i;
	for(int i = 1;i <= n;i++) {
		for(int j = 1;j <= n;j++) {
			scanf("%d",&p[i][j]);
			if(i < j) {
				e[++cnt].fr = i,e[cnt].to = j,e[cnt].we = p[i][j];
			}
		}
	}
	sort(e + 1,e + cnt + 1);
	int anss = n + 1;
	for(int i = 1;i <= cnt;i++) {
		if(find(e[i].fr) != find(e[i].to)) {
			//printf("%d %d %d\n",e[i].fr,e[i].to,e[i].we);
			head[find(e[i].to)] = find(e[i].fr);
			anss--;
			ans += e[i].we;
		}
		if(anss == 1) {
			printf("%d\n",ans);
			return 0;
		}
	}
    return 0;
}


```cpp