nowcoder 2018年全国多校算法寒假训练营练习比赛（第四场）-道路建设

链接：https://www.nowcoder.com/acm/contest/76/B
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 65536K，其他语言131072K
64bit IO Format: %lld

题目描述

随着如今社会的不断变化，交通问题也变得越来越重要，所以市长决定建设一些公路来方便各个城市之间的贸易和交易。虽然市长的想法很好，但是他也遇到了一般人也经常头疼的问题，那就是手头的经费有限……在规划过程中，设计师们已经预算出部分城市之间建设公路的经费需求。现在市长想知道，它能不能将他的m个城市在有限的经费内实现公路交通。如果可以的话，输出Yes，否则输出No（两个城市不一定要直接的公路相连，间接公路到达也可以。）

输入描述:

测试输入包含多条测试数据
每个测试数据的第1行分别给出可用的经费c(<1000000)，道路数目n(n<10000)，以及城市数目m(<100)。
接下来的n行给出建立公路的成本信息，每行给出三个整数，分别是相连的两个城市v1、v2(0<v1,v2<=m)以及建设公路所需的成本h(h<100)。

输出描述:

对每个测试用例，输出Yes或No。

示例1

输入

20 10 5
1 2 6
1 3 3
1 4 4
1 5 5
2 3 7
2 4 7
2 5 8
3 4 6
3 5 9
4 5 2

输出

Yes

示例2

输入

10 2 2
1 2 5
1 2 15

输出

Yes

备注:

两个城市之间可能存在多条线路

分析：

简单的最小生成树问题。

#include<bits/stdc++.h>
#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
#define INF 0x3f3f3f3f
int n,m,C;
int map1[105][105];

void prim()
{
    int low[105],vis[105];
    int bj,sum,min1,flag;
	memset(vis,0,sizeof(vis));
	vis[1]=1;
	bj=1;
	sum=0;///统计最少的成本
	flag=0;
	for(int i=2;i<=m;i++)
        low[i]=map1[bj][i];
	for(int i=1;i<m;i++)///注意这里的循环次数
	{
		min1= INF;
		for(int j=1;j<=m;j++)///找出最小的
		{
			if(!vis[j]&&min1>low[j])
			{
				min1=low[j];
				bj=j;
			}
		}
		sum+=min1;
		if(min1==INF)///如果最小的都为INF，说明没有可以找的了
        {
            flag=1;
            break;
        }
		vis[bj]=1;///将找过得点都标记
		for(int j=1;j<=m;j++)///更新low数组
		{
			if(!vis[j]&&low[j]>map1[bj][j])
				low[j]=map1[bj][j];
		}
	}
	if((!flag)&&(sum<=C))
        printf("Yes\n");
    else
        printf("No\n");
}
int main()
{
    int a,b,c;
	while(~scanf("%d %d %d",&C,&n,&m))
	{
		for(int i=0;i<=m;i++)
		{
			for(int j=0;j<=m;j++)
            {
                if(i==j)
                    map1[i][j]=0;
                else
                    map1[i][j]=INF;
            }
		}
		for(int i=1;i<=n;i++)
		{
		    scanf("%d %d %d",&a,&b,&c);
		    if(c<map1[a][b])
                map1[a][b]=map1[b][a]=c;
		}
        prim();
	}
}