求助
查看原帖
求助
862827
_LogicFish_楼主2025/1/25 22:40

就最后一个 AC\color{green}{AC} 了,为什么
我看了一下我的dfs()是没问题的,Kruskal()跑歪了(

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>

using namespace std;
#define int long long
int n,m,h[100005],x,y,z;
int cnt,sum;
struct node{
	int v,w;
};
struct Side{
	int u,v,w;
	bool operator <(const Side other) const
	{
		return (h[v]==h[other.v])?w<other.w:h[v]>h[other.v];
	}
};

int fa[100005];
bool bk[100005];
vector<node>mp[100005];
vector<Side>side;
int find(int);
void dfs(int);
void Krus();
signed main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=n;i++) cin>>h[i];
	for(int i=0;i<m;i++)
	{
		cin>>x>>y>>z;
		if(h[x]>=h[y])      mp[x].push_back({y,z});
		else if(h[x]<=h[y]) mp[y].push_back({x,z});
	}
	dfs(1);
	cout<<cnt<<' ';
	sort(side.begin(),side.end());
	Krus();
	cout<<sum;
	return 0;
}
int find(int x)
{
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
void dfs(int x)
{
	cnt++,bk[x]=1;
	for(auto i:mp[x])
	{
		side.push_back({x,i.v,i.w});
		if(!bk[i.v])
			dfs(i.v);
	}
}
void Krus()
{
	int num=0;
	for(auto i:side)
	{
		if(find(i.u)!=find(i.v))
		{
			num++,sum+=i.w,
			fa[find(i.u)]=find(i.v);
			if(num==cnt-1) return;
		}
	}
}

谢谢

2025/1/25 22:40
加载中...