CCF 2014-12-4 最优灌溉 100分

试题编号：	201412-4
试题名称：	最优灌溉
时间限制：	1.0s
内存限制：	256.0MB
问题描述：	问题描述 　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。 　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。 　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。 输入格式 　　输入的第一行包含两个正整数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。

 

/*---普利姆算法---*/
#include<iostream>
#include<algorithm>
using namespace std;

const int MAXV=1001;//最大顶点数 
const int INF=1000000000;
int n,m,G[MAXV][MAXV]={0};//G存储图 
int d[MAXV];  //顶点与集合S的最短距离 
bool vis[MAXV]={false};//访问标记 

int prim(){//默认从1号顶点开始，函数返回最小生成树的边权之和 
	fill(d,d+MAXV,INF);//fill将d数组改为INF 
	d[1]=0;//只有1号顶点到集合S的距离为0 
	int ans=0;
	for(int i=0;i<n;i++){//循环n次，i不需要从1开始 
		int u=-1,mn=INF;//u使d[u]最小，mn存放该最小的d[u] 
		for(int j=1;j<=n;j++){//找到为访问的顶点中的d[]最小的 
			if(vis[j]==false && d[j]<mn){
				u=j;
				mn=d[j];
			}
		}
		//找不到小于INF的d[u],则剩下的与集合S不连通 
		if(u==-1)return -1;
		vis[u]=true;//标记u为已访问 
		ans+=d[u];//将与集合S距离最小的边加入最小生成树边权 
		for(int v=1;v<=n;v++){
			//v未访问 && u能到达 && 以u为中介点可以使v离集合S更近 
			if(vis[v]==false && G[u][v]!=INF && G[u][v]<d[v]){
				d[v]=G[u][v];//将G[u][v]赋给d[v] 
			}
		}
	}
	
	return ans;
}

int main(){
	int u,v,w; 
	cin>>n>>m;
	fill(G[0],G[0]+MAXV*MAXV,INF);
	
	for(int i=0;i<m;i++){//输入图 
		cin>>u>>v>>w;
		G[u][v]=G[v][u]=w;
	}
	
	int ans=prim();
	cout<<ans;
}

更多相关CCF的试题解答，请点击>>CCF历年认证考试解答