牛客网 道路建设（prime+kruskal 算法）两种方法实现

什么prime算法 不就是迪杰特斯拉么....

牛客网上面无论考虑图是否连通都可以过

还有城市序号从1开始，我也踩了....

踩雷王就是我 我是酷炫的小菜鸡

题目链接：https://www.nowcoder.com/acm/contest/76/B

这是prime算法（考虑了图是否连通）

 /*
 qq:1239198605
 ctgu_yyf
        */

#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
int inf=99999;
int map[105][105],dis[105],vis[105];
int c,n,m; 
void prime(int x)
{
	memset(vis,0,sizeof(vis));
	vis[x]=1;

	
	for(int i=1;i<=m;i++)
	dis[i]=map[x][i];
    
	dis[x]=0; 
	
	for(int i=1;i<=m;i++)
	{
		if(i==x)
		continue;
		
		int min=inf;
		int u=-1;
		for(int j=1;j<=m;j++)
		{
			if(dis[j]<min&&vis[j]==0)
			{
				min=dis[j];
				u=j;
			}
		}
		if(u==-1)
		break;
		
		vis[u]=1;
		dis[u]=min;
		
		for(int z=1;z<=m;z++)
		if(vis[z]==0&&dis[u]+map[u][z]<dis[z])
		{
			dis[z]=dis[u]+map[u][z];
		}
			
	}
		
}


int main()
{
   ios::sync_with_stdio(false);
   
   int v1,v2,h;
   while(cin>>c>>n>>m)
   {
   	int flag=0;
   	for(int i=1;i<=m;i++)
   	for(int j=1;j<=m;j++)
   	map[i][j]=inf;
   	
   	
   	
   	for(int i=0;i<n;i++)
   	{
	   cin>>v1>>v2>>h;
	   map[v1][v2]=min(map[v1][v2],h);
	   map[v2][v1]=min(map[v2][v1],h);
   }
   
   prime(1);
   
   int ans=0;
   for(int i=1;i<=m;i++)
   {
   //	cout<<"**"<<dis[i]<<endl;
   	
   	if(dis[i]==inf)
   	flag=1;
   	ans=ans+dis[i];
   }
   
 //  cout<<"**"<<ans<<endl;
   
   
   if(ans>c)
   flag=1;
   
   if(flag==0)
   cout<<"Yes"<<endl;
   
   else
   cout<<"No"<<endl;
   	
   	
   }


return 0;
}

然后我试着用kruskal解一遍也过了

kruskal什么花里胡哨的东西，不就是个并查集么.....

然后是kruskal代码（没考虑图是否连通）：

 /*
 qq:1239198605
 ctgu_yyf
        */

#include<iostream>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
struct Edge{
	int u,v,cost;
}; 

Edge edge[10005];
int c,n,m;
int par[105];

void init(int n)
{
	for(int i=0;i<n;i++)
	par[i]=i;
}

int find(int x)
{
	if(par[x]==x)
	return x;
	
	else
	return par[x]=find(par[x]);
}

bool same(int a,int b)
{
	return find(a)==find(b);
}

bool cmp(const Edge &a,const Edge &b)
{
	return a.cost<b.cost;
}

void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	
	if(x==y)
	return ;
	
	par[y]=x;
	
}

int kruskal()
{
	sort(edge,edge+n,cmp);
	init(m);
	int res=0;
	
	for(int i=0;i<n;i++)
	{
		if(same(edge[i].u,edge[i].v)==false)
		{
		unite(edge[i].u,edge[i].v);
		res=res+edge[i].cost;
	    }
	}
	
	return res;
}

int main()
{
   ios::sync_with_stdio(false);
   
   
   while(cin>>c>>n>>m)
   {
   
   
   for(int i=0;i<n;i++)
   cin>>edge[i].u>>edge[i].v>>edge[i].cost;
   
   if(kruskal()<=c)
   cout<<"Yes"<<endl;
   else
   cout<<"No"<<endl;
   
   
   }

return 0;
}

两个都可以过。