CSP201412第四题--最优灌溉(C++)

题目

题目链接

link.

问题描述

雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号
接下来m行，每行包含三个整数ai, bi, ci，表示第ai片麦田与第bi片麦田之间可以建立一条水渠，所需要的费用为ci。

输出格式

输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

样例说明

建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。

评测用例规模与约定

前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤ci≤10,000。

解题思路

一道普通的最小生成树问题…
一开始我还在想如果图不连通怎么解决，但这道题的测试全部是连通图，那就简单了。另外样例规模可以看到边数多于点数，使用prim算法时间效率比较高。

算法	代码长度	时间使用	空间使用
prim	759B	156ms	4.277MB
kurskal	870B	171ms	1.945MB

代码

prim

#include<stdio.h>
#include<string.h>
const int maxn=1e3+5;
const int inf=0x3f3f3f3f;
int n,m;
int d[maxn];          //点和当前集合距离 
bool vis[maxn]={
   0};   //访问状态 
int dis[maxn][maxn];  //灌溉费用 

int prim(){
   
	int ans=0;        //费用 
	d[1]=0;           //第一个点和当前集合（空集）的距离（费用）初始化为0 
	for(int i=1;i<=n;i++){
   
		int min=inf,k=-1;     //寻找和当前集合最近的点 
		for(int j