萌新刚学OI,改数小时未果
查看原帖
萌新刚学OI,改数小时未果
222104
_yjh楼主2020/8/6 23:12

求各位大佬帮忙查一下错误,刚学图论和倍增,自己也找不出错误QAQ

题目链接:点我

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
using namespace std;
long long n,m,cnt,adr,ans,ANS=2147483647,f[100050][26],g[100050][26],depth[100050];
vector <long long> son[100050];
struct Edge {
	long long x,y,q;
}e[300050]; 
struct Bug {
	int f,q;
}fa[100050]; 
bool operator < (Edge x,Edge y) {
	return x.q<y.q;
}
long long get_father(long long x) {
	if(x==fa[x].f) {
		return x;
	}
	return fa[x].f=get_father(fa[x].f);
}
void Merge(long long x,long long y) {
	int fx=get_father(x),fy=get_father(y);
	if(fx!=fy) {
		fa[fx].f=fy;
		return ;
	}
	return ;
}
void dfs(int now) {
	for(int i=0;i<son[now].size();i++) {
		depth[i]=depth[now]+1;
		for(int j=1;j<=25;j++) {
			f[i][j]=f[f[i][j-1]][j-1]+f[i][j-1];
			g[i][j]=max(g[i][j-1],g[f[i][j-1]][j-1]);
		}
		dfs(i);
	}
}
long long LCA(int p1,int p2,int x) {
	if(depth[p1]<depth[p2]) {
		swap(p1,p2);
	}
	long long ans=-1;
	for(long long i=25;i>=0;i--) {
		if(depth[f[p1][i]]>=depth[p2]) {
			if(g[p1][i]!=e[x].q) {
			    ans=max(ans,g[p1][i]);
		    }
			p1=f[p1][i];
		}
	}
	if(p1==p2) {
		return ans;
	}
	for(int i=25;i>=0;i--) {
		if(f[p1][i]!=f[p2][i]) { 
		    if(g[p1][i]!=e[x].q&&g[p2][i]!=e[x].q) {
		    	ans=max(ans,max(g[p1][i],g[p2][i]));
			}
			else  if(g[p1][i]!=e[x].q) {
				ans=max(ans,g[p1][i]);
			}
			    else {
			    	ans=max(ans,g[p2][i]);
				}
			p1=f[p1][i];
			p2=f[p2][i];
		}
	}
	return max(ans,max(g[p1][0],g[p2][0]));
}
int main() {
	scanf("%lld%lld",&n,&m);
	for(long long i=1;i<=m;i++) {
		scanf("%lld%lld%lld",&e[i].x,&e[i].y,&e[i].q);
	}
	for(int i=1;i<=n;i++) {
		fa[i].f=i;
	}
	sort(e+1,e+m+1);
	for(int i=1;i<=m;i++) {
		long long fx=get_father(e[i].x),fy=get_father(e[i].y);
		if(fx!=fy) {
			long long emax=max(e[i].x,e[i].y),emin=min(e[i].x,e[i].y);
			Merge(emax,emin);
			son[emin].push_back(emax);
			fa[emax].f=emin;
			fa[emax].q=e[i].q;
			ans+=e[i].q; 
			++cnt;
		}
		if(cnt==n-1) {
			adr=i+1;
			break;
		}
	}
	depth[1]=1;
	for(long long i=1;i<=n;i++) {
		f[i][0]=fa[i].f;
		g[i][0]=fa[i].q;
	}
	dfs(1);
	for(long long i=adr;i<=m;i++) {
		ANS=min(ANS,e[i].q-LCA(e[i].x,e[i].y,i));
	}
	printf("%lld",ans+ANS);
	return 0;
}
2020/8/6 23:12
加载中...