爆0!!!
查看原帖
爆0!!!
939931
Addicted_Game楼主2023/11/3 18:53
#include<bits/stdc++.h>
using namespace std;
int n,q,m;
int h[1000005],e[1000005],ne[1000005],w[1000005],cnt;
bool flag[200005];
struct node{
	int x,y,w;
}edg[1000005];
void add(int x,int y,int z){
	cnt++;
	e[cnt]=y;
	ne[cnt]=h[x];
	h[x]=cnt; 
	w[cnt]=z;
}
int depth[200005];
int f[200005][25];
int road[200005];
int k;
int fa[200005];
bool cmp(node a,node b){
	return a.w>b.w;
} 
void init(){//初始化 
	for(int i=1;i<=n;i++) fa[i]=i;
} 
int find(int x){//查询根节点 
	if(fa[x]!=x) return fa[x]=find(fa[x]);
	return x; 
}
bool same_set(int x,int y){//判断是否在一个集合 
	x=find(x);
	y=find(y);
	return x==y;
}
void union_set(int x,int y){//合并所在的集合 
	x=find(x);
	y=find(y);
	fa[x]=y;
}
void dfs(int u,int fa){
	depth[u]=depth[fa]+1;
	f[u][0]=fa;
	for(int j=1;;j++){
		f[u][j]=f[f[u][j-1]][j-1];
		if(!f[u][j]){
			k=max(k,j-1);
			break; 
		}
	}
	for(int j=h[u];j!=-1;j=ne[j]){
		int to=e[j];
		if(to!=fa){
			road[to]=road[u]+w[j];
			dfs(to,u);
		}
	}
}
int get_lca(int u,int v){
	if(depth[u]>depth[v]) swap(u,v);
	for(int j=k;j>=0;j--){
		if(depth[u]<=depth[f[v][j]]){
			v=f[v][j];
		}
	}
	if(u==v) return u;
	for(int j=k;j>=0;j--){
		if(f[u][j]!=f[v][j]){
			u=f[u][j];
			v=f[v][j];
		}
	}
	return f[u][0];
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	memset(h,-1,sizeof h);
	cin>>n>>m;
	for(int i=1;i<=n-1;i++){
		cin>>edg[i].x>>edg[i].y>>edg[i].w;
	}
	init();
	sort(edg+1,edg+m+1,cmp);
	int t=0;
	for(int i=1;i<=m;i++){
		int x=edg[i].x;
		int y=edg[i].y;
		int w=edg[i].w;
		if(!same_set(x,y)){
			//printf("%d %d\n",x,y);
			add(x,y,w);
			add(y,x,w);
			flag[y]=true;
			t++;
			if(t>=n-1) break;
		} 
	}
	cin>>q;
	int t2=0;
	for(int i=1;i<=n;i++){
		if(!flag[i]){
			t2=i;
			break;
		}
	}
	dfs(t2,0);
	while(q--){
		int l,r,t1;
		cin>>l>>r;
		if(same_set(l,r)) cout<<"-1"<<endl;
		else{
			t1=get_lca(l,r);
			cout<<road[l]+road[r]-2*road[t1]<<endl;
		}
		
	}
	//for(int i=1;i<=n;i++) cout<<road[i]<<endl;
	return 0;
}

码了老半天的缝合怪结果样例爆零,WA+RE

2023/11/3 18:53
加载中...