求助,整体感觉思路清晰但全WA!
查看原帖
求助,整体感觉思路清晰但全WA!
475329
Larryyu楼主2022/8/4 11:08
#include<bits/stdc++.h>
using namespace std;
#define N 100100
#define M 200200
int n,m;
long long tot,cnt;
unsigned long long sum,mst=999999999999999ull;
long long fa[N],head[N];
long long id[N][20],maxv[N][20],secv[N][20],dep[N];
struct edge{
	long long x;
	long long y;
	long long w;
	bool vis;
}e[M];
struct tree{
	long long to;
	long long next;
	long long w;
}tre[M];
long long find(long long x){
	if(fa[x]==x) return fa[x];
	return fa[x]=find(fa[x]);
}
void add(long long x,long long y,long long z){
	tre[++tot].to=y;
	tre[tot].w=z;
	tre[tot].next=head[x];
	head[x]=tot;
}
bool cmp(edge x,edge y){
	return x.w<y.w;
}
void dfs(long long x,long long fx){
	for(long long i=head[x];i;i=tre[i].next){
		long long y=tre[i].to;
		if(y==fx) continue;
		dep[y]=dep[x]+1;
		maxv[y][0]=tre[i].w;
		id[y][0]=x;
		dfs(y,x);
	}
}
long long getl(long long x,long long y){
	if(dep[x]<dep[y]){
		swap(x,y);
	}
	for(long long i=20;i>=0;i--){
		if(dep[id[x][i]]>=dep[y]){
			x=id[x][i];
		}
	}
	if(x==y) return x;
	for(long long i=20;i>=0;i--){
		if(id[x][i]!=id[x][i]){
			x=id[x][i];
			y=id[y][i];
		}
	}
	return id[x][0];
}
long long get(long long x,long long y,long long w){
	long long lca=getl(x,y),l=0,r=0;
	/*for(long long i=0;i<=20;i++){
		if((1<<i)>dep[x]){
			break;
		}
	}*/
	for(long long i=20;i>=0;i--){
		if(dep[id[x][i]]>=dep[lca]){
			if(maxv[x][i]!=w){
				l=max(l,maxv[x][i]);
			}else{
				l=max(l,secv[x][i]);
			}
			x=id[x][i];
		}
	}
	for(long long i=20;i>=0;i--){
		if(dep[id[y][i]]>=dep[lca]){
			if(maxv[y][i]!=w){
				r=max(r,maxv[y][i]);
			}else{
				r=max(r,secv[y][i]);
			}
			r=id[r][i];
		}
	}
	return max(l,r);
}
int main(){
	cin>>n>>m;
	for(long long i=1;i<=n;i++){
		fa[i]=i;
	}
	for(long long i=1;i<=m;i++){
		cin>>e[i].x>>e[i].y>>e[i].w;
	}
	sort(e+1,e+1+m,cmp);
	for(long long i=1;i<=m;i++){
		long long fx=find(e[i].x);
		long long fy=find(e[i].y);
		if(fx==fy) continue;
		cnt++;
		fa[fy]=fx;
		e[i].vis=1;
		sum+=e[i].w;
		add(e[i].x,e[i].y,e[i].w);
		add(e[i].y,e[i].x,e[i].w);
		if(cnt==n-1) break;
	}
	dfs(1,0);
	for(long long i=1;i<=18;i++){
		for(long long j=1;j<=n;j++){
			id[j][i]=id[id[j][i-1]][i-1];
			maxv[j][i]=max(maxv[j][i-1],maxv[id[j][i-1]][i-1]);
			secv[j][i]=max(secv[j][i-1],secv[id[j][i-1]][i-1]);
			if(maxv[j][i-1]>maxv[id[j][i-1]][i-1]&&maxv[id[j][i-1]][i-1]>secv[j][i]){
				secv[j][i]=maxv[id[j][i-1]][i-1];
			}else if(maxv[id[j][i-1]][i-1]>maxv[j][i-1]&&maxv[j][i-1]>secv[j][i]){
				secv[j][i]=maxv[j][i-1];
			}
		}
	}
	for(long long i=1;i<=m;i++){
		if(!e[i].vis){
			long long ans=get(e[i].x,e[i].y,e[i].w);
			if(ans!=e[i].w&&mst>sum-ans+e[i].w){
				mst=sum-ans+e[i].w;
			}
		}
	}
	cout<<mst<<endl;
	return 0;
}
2022/8/4 11:08
加载中...