BZOJ3631【LCT】

写的LCT.跑起来好慢好慢.无限仰慕用Tree DP秒题的菊苣.

/* I will wait for you */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iostream>
#include<fstream>
#include<vector>
#include<queue>
#include<deque>
#include<set>
#include<map>
#include<string>
#define make make_pair
#define fi first
#define se second

using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int maxn=500010;
const int maxm=1010;
const int maxs=26;
const int INF=1<<29;
const int P=1000000007;
const double error=1e-9;

struct node
{
	int key,add,rev;
	node *fa,*son[2];
	
	int dir() { return this==fa->son[1]; }
	int isroot() { return !fa||this!=fa->son[0]&&this!=fa->son[1]; }
}no[maxn],*su[maxn];

void init(node* &o)
{
	o->key=o->add=o->rev=0;
	o->fa=o->son[0]=o->son[1]=0;
}

void pushdown(node* o)
{
	if(o->rev)
	{
		o->rev=0;swap(o->son[0],o->son[1]);
		for(int i=0;i<2;i++) if(o->son[i]) o->son[i]->rev^=1;
	}
	if(o->add)
	{
		for(int i=0;i<2;i++) if(o->son[i]) o->son[i]->add+=o->add;
		o->key+=o->add;o->add=0;
	}
}

void maintain(node* o)
{
	for(int i=0;i<2;i++) if(o->son[i]) o->son[i]->fa=o;
}

void rotate(node* o)
{
	node* p=o->fa;
	pushdown(p);pushdown(o);
	int d=o->dir();
	p->son[d]=o->son[d^1];
	o->son[d^1]=p;o->fa=p->fa;
	if(!p->isroot()) p->fa->son[p->dir()]=o;
	maintain(p);maintain(o);
}

void splay(node* o)
{
	pushdown(o);
	while(!o->isroot())
	{
		node* p=o->fa;
		if(p->isroot()) rotate(o);
		else if(o->dir()==p->dir()) rotate(p),rotate(o);
		else rotate(o),rotate(o);
	}
}

void access(node* o)
{
	for(node* t=0;o;t=o,o=o->fa) splay(o),o->son[1]=t,maintain(o);
}

void evert(node* o)
{
	access(o);splay(o);o->rev^=1;
}

void link(node* u,node* v)
{
	evert(u);u->fa=v;
}

void ins(node* u,node* v)
{
	evert(u);access(v);splay(v);v->add++;
}

int query(node* o)
{
	splay(o);pushdown(o);return o->key;
}

int c[maxn];

int main()
{
	int n;scanf("%d",&n);
	
	for(int i=1;i<=n;i++) scanf("%d",&c[i]),su[i]=&no[i],init(su[i]);
	for(int i=1,u,v;i<n;i++) scanf("%d%d",&u,&v),link(su[u],su[v]);
	for(int i=1;i<n;i++) ins(su[c[i]],su[c[i+1]]);
	for(int i=1;i<=n;i++) printf("%d\n",query(su[i])-(i!=c[1]));
	
	return 0;
}