过了样例,WA#1剩下RE,求助(玄关)
查看原帖
过了样例,WA#1剩下RE,求助(玄关)
1258467
hao12345ia楼主2025/2/8 16:53
#include<bits/stdc++.h>
using namespace std;
struct lin{
	int cnt,to;
};
struct edge{
	int l,r,val;
}e[1000001],s[1000001];
vector<lin> a[1000001];
int vis[1000001];
long long n,cnt,m,a1[100001];
void dfs(int now){
	vis[now]=1;
	for(lin next:a[now]){
		int to=next.to;
		if(!vis[to]){
			dfs(to);
			s[++cnt]=e[next.cnt];
		}
	}
	vis[now]=0;
}
bool cmp(edge x,edge y){
	if(a1[x.r]==a1[y.r]){
		return x.val<y.val;
	}
	return a1[x.r]>a1[y.r];
}
int fa[10000001];
int find(int x){
	if(fa[x]==x){
		return x;
	}
	return fa[x]=find(fa[x]);
}
void join(int x,int y){
	x=find(x),y=find(y);
	fa[x]=fa[y];
}
bool vis1[1000001];
int ku(){
	int ans=0,ans1=0;
	for(int i=1;i<=cnt;i++){
		int a=find(s[i].l);
		int b=find(s[i].r);
		if(a!=b){
		//	cout<<s[i].l<<" "<<s[i].r<<endl;
			ans+=s[i].val;
			ans1+=(1^vis1[a])+(1^vis1[b]);
			vis1[a]=vis1[b]=1;
			join(a,b);
		}
	}
	//cout<<ans1<<" "<<ans;
	return 0;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		fa[i]=i;
		cin>>a1[i];
	}
	for(int i=1;i<=m;i++){
		int l,r,val;
		cin>>l>>r>>val;
		if(a1[l]<a1[r]){
			swap(l,r);
		}
		e[i]={l,r,val};
		a[l].push_back({i,r});
	}
	dfs(1);
	sort(s+1,s+1+cnt,cmp);
	ku();
}
2025/2/8 16:53
加载中...