萌新求助次小生成树,10pts
查看原帖
萌新求助次小生成树,10pts
365532
Mr_ll楼主2022/11/5 14:19

照着attack大佬的题解写的

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e5+10,M=3e5+10;
int n,m,bcj[N],cnt,hea[M<<1],to[M<<1],net[M<<1],dis[N<<1],dep[N],f[N][25],mx[N][25],me[N][25],tot;
long long sum,ans=1e16;
bool vis[N];
int read() {
	int x=0,f=1;char ch=getchar();
	while((ch<'0'||ch>'9')&&(ch!='-')) ch=getchar();
	if(ch=='-') f=-1,ch=getchar();
	while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x*f;
}
struct nodeE {
	int x,y,z;
	bool operator <(const nodeE b)const {
		return z<b.z;
	}
}E[M];
int find(int x) {
	return bcj[x]==x?x:bcj[x]=find(bcj[x]);
}
void add(int x,int y,int z) {
	to[++tot]=y;
	dis[tot]=z;
	net[tot]=hea[x];
	hea[x]=tot;
}
void kru() {
	sort(E+1,E+1+n);
	for(int i=1;i<=n;i++) bcj[i]=i;
	for(int i=1;i<=m;i++) {
		int x=E[i].x,y=E[i].y;
		int fx=find(x),fy=find(y);
		if(fx==fy) continue;
		bcj[max(fx,fy)]=min(fx,fy);
		sum+=(long long)E[i].z;
		add(x,y,E[i].z);add(y,x,E[i].z);
		cnt++;vis[i]=1;
		if(cnt==n-1) break;
	} 
}
void dfs(int x,int fa) {
	dep[x]=dep[fa]+1;
	f[x][0]=fa;
	for(int i=hea[x];i;i=net[i]) {
		int y=to[i];
		if(y==fa) continue;
		mx[y][0]=dis[i];
		dfs(y,x);
	}
}
void pre() {
	for(int j=1;j<=22;j++) {
		for(int i=1;i<=n;i++) {
			f[i][j]=f[f[i][j-1]][j-1];
			mx[i][j]=max(mx[i][j-1],mx[f[i][j-1]][j-1]);
			me[i][j]=max(me[i][j-1],me[f[i][j-1]][j-1]);
			//yi
			if(mx[i][j-1]>mx[f[i][j-1]][j-1]) me[i][j]=max(me[i][j],mx[f[i-1][j-1]][j-1]);
			else if(mx[i][j-1]<mx[f[i][j-1]][j-1]) me[i][j]=max(me[i][j],mx[i][j-1]);
 		}
	}
}

int lca(int x,int y) {
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=22;i>=0;i--) 
		if(dep[f[x][i]]>=dep[y]) x=f[x][i]; 
	if(x==y) return x;
	for(int i=22;i>=0;i--)
		if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}

int get_mx(int x,int lc,int val) {
	long long ans=0;
	for(int i=22;i>=0;i--) {
		if(dep[f[x][i]]>=dep[lc]) {
			if(val!=mx[x][i]) ans=max(ans,(long long)mx[x][i]);
			else ans=max(ans,(long long)me[x][i]); 
			x=f[x][i];
		} 
	}
	return ans;
}

void work() {
	for(int i=1;i<=m;i++) {
		if(vis[i]) continue;
		int x=E[i].x,y=E[i].y,z=E[i].z;
		int lc=lca(x,y);
		int lmx=get_mx(x,lc,z),rmx=get_mx(y,lc,z);
		if(max(lmx,rmx)!=z) ans=min(ans,sum+z-max(lmx,rmx));
	}
}


int main() {
	n=read();m=read();
	for(int i=1;i<=n;i++) E[i].x=read(),E[i].y=read(),E[i].z=read();
	kru();
	dfs(1,0);
	pre();
	work();
	printf("%lld\n",ans);
	return 0; 
}
2022/11/5 14:19
加载中...