vector 可以实现trajanLCA 吗?
  • 板块灌水区
  • 楼主不慕放糖
  • 当前回复3
  • 已保存回复3
  • 发布时间2021/10/16 10:38
  • 上次更新2023/11/4 03:41:19
查看原帖
vector 可以实现trajanLCA 吗?
544113
不慕放糖楼主2021/10/16 10:38
#include<bits/stdc++.h>
using namespace std;
int fa[5000000],ans[600000];
int a[5000000],b[5000000];
bool vis[5000000];	
int n,m,s;
int cnt=0;
int find(int x){
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
void uni(int x,int y){
	int a=find(x);
	int b=find(y);
	if(a!=b)   fa[a]=b;
}
struct qustion{
	int a,b;
}z[1000000];
vector<int>q[1000000];
vector<int>u[1000000];
void dfs(int x){
	vis[x]=1;
	for(int i=0;i<q[x].size();i++){
		if(!vis[q[x][i]])       dfs(q[x][i]), uni(q[x][i],x);
	}
	
	for(int i=0;i<u[x].size();i++){
		int y=u[x][i];
		if(vis[y]) {
			for(int i=1;i<=m;i++){
				if((z[i].a==x&&z[i].b==y)||z[i].a==y&&z[i].b==x)  if(find(y)==0)ans[i]=find(x); else ans[i]=find(y);		
			}
		}
	}
}
int main(){

	cin>>n>>m>>s;
	for(int i=1;i<=n;i++) fa[i]=i; 
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		q[u].push_back(v);
		q[v].push_back(u);
	}
	for(int i=1;i<=m;i++){
		cin>>a[i]>>b[i];
		u[a[i]].push_back(b[i]); 
		u[b[i]].push_back(a[i]);
		z[i].a=a[i],z[i].b=b[i];
	}
	dfs(s);
	for(int i=1;i<=m;i++){
		cout<<ans[i]<<endl;
	}
	return 0;
}

顺利把从RE到MLE到TLE 求大佬调,已经写了一个上午了

2021/10/16 10:38
加载中...