就最后一个 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;
}
}
}
谢谢