玄关求调,样例过了但全wa(lca+树上差分)
查看原帖
玄关求调,样例过了但全wa(lca+树上差分)
946180
weiy_楼主2024/7/12 11:38
#include<bits/stdc++.h>
using namespace std;
const int maxn=300005;
int n;
int deep[maxn];
int head[maxn<<1],cnt;
int par[maxn][35],diff[maxn];
int ans[maxn];
int zc[maxn];
struct node{
	int next,to;
}e[maxn<<1];
void add(int a,int b){
	e[++cnt].to=b;
	e[cnt].next=head[a];
	head[a]=cnt;
}
void dfs(int x,int fa){
	deep[x]=deep[fa]+1;
	par[x][0]=fa;
	for(int i=head[x];i;i=e[i].next){
		int y=e[i].to;
		if(y!=fa){
			dfs(y,x);
		}
	}
}
int lca(int u,int v){
	if(deep[u]<deep[v]) swap(u,v);

	for(int j=31;j>=0;--j){
		if(deep[u]-(1<<j)>=deep[v])
			u=par[u][j];
	}
	if(u==v) return u;
	for(int j=31;j>=0;--j){
		if(par[u][j]!=par[v][j]){
			u=par[u][j];
			v=par[v][j];
		}
	}
	return par[u][0];
}
int get_ans(int x){
	ans[x]=diff[x];
	for(int i=head[x];i;i=e[i].next){
		int y=e[i].to;
		if(y!=par[x][0]){
			ans[x]+=get_ans(y);
		}
	}
	return ans[x];
}
int main(){
//	freopen("P3258_1.in","r",stdin);
//	freopen("P3258.out","w",stdout);
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>zc[i];
	}
	for(int i=1;i<n;i++){
		int a,b;
		cin>>a>>b;
		add(a,b);
		add(b,a);
	}
	dfs(1,0);
	for(int j=1;j<=31;j++){
		for(int i=1;i<=n;i++){
			par[i][j]=par[par[i][j-1]][j-1];
		}
	}
	for(int i=1;i<n;i++){
		int x=zc[i],y=zc[i+1];
		diff[x]+=1;
		diff[y]+=1;
		diff[lca(x,y)]-=1;
		diff[par[lca(x,y)][0]]-=1;
	}
	int ZCHAKIOI=get_ans(1);
	for(int i=1;i<=n;i++){
		cout<<ans[i]<<"\n";
	} 
	return 0;
}
2024/7/12 11:38
加载中...