试题编号: | 201412-4 |
试题名称: | 最优灌溉 |
时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。 输入格式 输入的第一行包含两个正整数n, m,分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。 输出格式 输出一行,包含一个整数,表示灌溉所有麦田所需要的最小费用。 样例输入 4 4 样例输出 6 样例说明 建立以下三条水渠:麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。 评测用例规模与约定 前20%的评测用例满足:n≤5。 |
/*---普利姆算法---*/
#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历年认证考试解答