Week6 C - 掌握魔法の东东 I

Week6 C - 掌握魔法の东东 I

东东在老家农村无聊，想种田。农田有 n 块，编号从 1~n。种田要灌氵
众所周知东东是一个魔法师，他可以消耗一定的 MP 在一块田上施展魔法，使得黄河之水天上来。他也可以消耗一定的 MP 在两块田的渠上建立传送门，使得这块田引用那块有水的田的水。 (1<=n<=3e2)
黄河之水天上来的消耗是 Wi，i 是农田编号 (1<=Wi<=1e5)
建立传送门的消耗是 Pij，i、j 是农田编号 (1<= Pij <=1e5, Pij = Pji, Pii =0)
东东为所有的田灌氵的最小消耗

Input

第1行：一个数n
第2行到第n+1行：数wi
第n+2行到第2n+1行：矩阵即pij矩阵

Output

东东最小消耗的MP值

Example
Input

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

Output

9

解题思路

黄河之水天上来可以视为个额外的农田 花费为到其他农田的路径花费
在对所有农田+新的农田(黄河农田)求最小生成树
先把每个边以权值 sort
利用 kruskal 算法 将边和并 累加路径长

Code

#include<iostream>
#include<iostream>
#include<vector>
#include<string.h>
#include<algorithm>
using namespace std;
int n;
int s[100000];
int rs[100000];
int t=1;
int ans=0;
void init(int x){
	for(int i=0;i<x;i++){
		s[i]=i;rs[i]=1;
	}
}
int find(int x){
	if(s[x]==x) return x;
	return find(s[x]);
}
bool unite(int x,int y){
	x=find(x);y=find(y);
	if(x==y)return false;
	if(rs[x]>rs[y]){s[y]=x;rs[x]=(rs[y]+=rs[x]);
	}else{s[x]=y;rs[y]=(rs[x]+=rs[y]);
	}
	return true;
}
struct Edge{
	int a;
	int b;
	int v=-1;
}edge[100000];

void kruskal(){
	int k=0;
	for(int i=0;i<t;i++){
		if(unite(edge[i].a,edge[i].b)){
			k++;
			ans+=edge[i].v;
		}
		if(k==n)return;
	}
}
bool cmp(Edge a,Edge b){
	return a.v<b.v;
}
int main(){
	
    cin>>n;
	init(n+1);
    int w;
    for(int i=1;i<=n;i++){
    	cin>>w;
    	edge[t].a=0;
    	edge[t].b=i;
    	edge[t].v=w;
    	t++;
	}
	for(int i=1;i<=n;i++){
    	for(int y=1;y<=n;y++){
    		cin>>w;
    		edge[t].a=i;
    		edge[t].b=y;
    		edge[t].v=w;
    		t++;
		}
	}
	sort(edge+1,edge+t+1,cmp);
	kruskal();
	cout<<ans;
}