黄河之水天上来——最小生成树

问题描述

东东在老家农村无聊，想种田。农田有 n 块，编号从 1~n。种田要灌氵
众所周知东东是一个魔法师，他可以消耗一定的 MP 在一块田上施展魔法，使得黄河之水天上来。他也可以消耗一定的 MP 在两块田的渠上建立传送门，使得这块田引用那块有水的田的水。 (1<=n<=3e2)
黄河之水天上来的消耗是 Wi，i 是农田编号 (1<=Wi<=1e5)
建立传送门的消耗是 Pij，i、j 是农田编号 (1<= Pij <=1e5, Pij = Pji, Pii =0)
东东为所有的田灌氵的最小消耗
【输入】
第一行：一个数n
第二行到第n+1行：wi
第n+2行到第2n+1行：矩阵即pij矩阵
【输出】
东东最下消耗的MP值
【样例输入】
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
【样例输出】
9

问题分析

如果没有黄河之水天上来，我们需要找的就是连接田地的最短路，即最小生成树。
有黄河之水天上来，我们可以理解为，多了一个田，可以与所有的田地连接，所以本题通过寻找n+1个田地的最小生成树解题，本题采用kruskal算法（不断选最短边，并用并查集判断法）
关于并查集，可以看上一篇文章，里有详解
关于kruskal的详解

代码块错误

如果有n个点，则最小生成树最终是选择n-1条边，因此一开始写选边操作时将for循环中的判断条件次数上限值设为边数，最后发现不是按顺序的所有最小边都可以被加入，有的可能成环无法加入，这就是并查集的作用。

#include<iostream>
#include<algorithm> 
using namespace std;

int n;
int fa[310];

struct edge
{
	int x,y,w;

}edges[100000];
bool compare(edge p,edge q)
{
	return p.w<q.w;
}
 
int find(int x)
{
	if(fa[x]==x) return x;
	else return fa[x]=find(fa[x]);
}

void add(int x,int y)
{
	int fx=fa[x];
	int fy=fa[y];
	fa[fx]=fy;
}

int main()
{
	cin>>n;
	int wi,p,sum=0;
	for(int i=1;i<=n;i++)
	{
		fa[i]=i;
	}
	for(int i=1;i<=n;i++)
	{
		cin>>wi;
		edges[i].x=i;
		edges[i].y=n+1;
		edges[i].w=wi;
	}
	int m=n;
	for(int j=1;j<=n;j++)
	{
		for(int k=1;k<=n;k++)
		{
			cin>>p;
			if(j!=k)
			{
				edges[++m].x=j;
				edges[m].y=k;
				edges[m].w=p; 
			} 
		}
	}
	sort(edges+1,edges+m,compare);
	int cnt=0;
    for(int k=1;k<m;k++)//未必能选n条边，报错点 
    {
    	int a=find(edges[k].x);
		int b=find(edges[k].y);
		if(cnt==n) break;
		if(a!=b) 
		{
			add(a,b); 
			sum+=edges[k].w;
			cnt++;
		}
	}
	cout<<sum<<endl;
}