问题描述
试题编号: | 201412-4 |
试题名称: | 最优灌溉 |
时间限制: | 1.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 雷雷承包了很多片麦田,为了灌溉这些麦田,雷雷在第一个麦田挖了一口很深的水井,所有的麦田都从这口井来引水灌溉。 输入格式 输入的第一行包含两个正整数n, m,分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。 输出格式 输出一行,包含一个整数,表示灌溉所有麦田所需要的最小费用。 样例输入 4 4 样例输出 6 样例说明 建立以下三条水渠:麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。 评测用例规模与约定 前20%的评测用例满足:n≤5。 |
思路:求最小生成树
AC代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <cstring>
#include <climits>
#include <cmath>
#include <cctype>
const int inf = 0x3f3f3f3f;//1061109567
typedef long long LL;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
int map1[1010][1010];
int lowdist[1010];
int visit[1010];
int n;
void prim()
{
for(int i=1; i<=n; i++)
lowdist[i] = map1[1][i];
visit[1] = 1;
int sum = 0;
int k;
for(int i=1; i<n; i++)
{
int min1 = inf;
for(int j=1; j<=n; j++)
{
if(!visit[j] && lowdist[j] < min1)
{
min1 = lowdist[j];
k = j;
}
}
visit[k] = 1;
sum += min1;
for(int j=1; j<=n; j++)
{
if(!visit[j] && map1[k][j] < lowdist[j])
lowdist[j] = map1[k][j];
}
}
printf("%d\n",sum);
}
int main()
{
int m;
scanf("%d%d",&n,&m);
memset(map1,inf,sizeof(map1));
memset(visit,0,sizeof(visit));
for(int i=0; i<m; i++)
{
int st,ed,value;
scanf("%d%d%d",&st,&ed,&value);
if(map1[st][ed] > value)
map1[st][ed] = map1[ed][st] = value;
}
prim();
return 0;
}