什么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;
}
两个都可以过。