老八老八,谁的一辈子
  • 板块灌水区
  • 楼主Ghosty_Neutrino
  • 当前回复1
  • 已保存回复1
  • 发布时间2024/12/28 16:47
  • 上次更新2024/12/28 20:43:20
查看原帖
老八老八,谁的一辈子
1082298
Ghosty_Neutrino楼主2024/12/28 16:47
#include <bits/stdc++.h>
using namespace std;
const int N=3e5+10,L=20;
struct D{
	int to,next;
}a[N*2];
int k,pre[N],v[N],lg[N],fa[N][L],dep[N],n,x,y,cnt[N];
void add(int u,int v){
	a[++k]={v,pre[u]};
	pre[u]=k;
}
void dfs(int x,int fath){
	fa[x][0]=fath;
	dep[x]=dep[fath]+1;
	for(int i=pre[x];i;i=a[i].next){
		int to=a[i].to;
		if(to!=fath){
			dfs(to,x);
			cnt[x]+=cnt[to];
		}
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	while(dep[x]!=dep[y]) x=fa[x][lg[dep[x]-dep[y]]];
	if(x==y) return x;
	for(int i=L-1;i>=0;i--){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&v[i]);
	int x,y;
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	lg[1]=0;
	for(int i=2;i<=n;i++) lg[i]=lg[i/2]+1;
	dfs(1,0);
	for(int j=1;j<L;j++){
		for(int i=1;i<=n;i++) fa[i][j]=fa[fa[i][j-1]][j-1];
	}
	int r;
	for(int i=1;i<n;i++){
		x=v[i];
		y=v[i+1];
		r=lca(x,y);
		cnt[x]++,cnt[y]++,cnt[r]--;
		cnt[fa[r][0]]--;
	}
	dfs(1,0);
	for(int i=2;i<=n;i++) cnt[v[i]]--;
	for(int i=1;i<=n;i++){
		printf("%d\n",cnt[i]);
	}
	return 0;
}

几百年前的省选题搞得老八背了一半卡住了

2024/12/28 16:47
加载中...