90pts,#4TLE求助
查看原帖
90pts,#4TLE求助
228745
AllenKING_RED楼主2022/8/17 10:57

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+9;
int n,a[N],id[N],top[N],son[N],fa[N],siz[N],dep[N],cnt;
struct node{
	int l,r;
	int add,sum;
}tr[N<<2];

vector<int>edge[N];
inline int read()
{
    int s=0,f=1;
    char ch=getchar();
    while (!isdigit(ch))
    {
        ch=='-'?f=-1:f=1;
        ch=getchar();
    }
    while (isdigit(ch))
    {
        s=(s<<3)+(s<<1)+(ch^48);
        ch=getchar();
    }
    return s*f;
}
inline void write(int x)
{
    (x<0)?(putchar('-'),x=-x):-1;
    if(x>9)write(x/10);
    putchar(x%10+'0');
}
void dfs1(int u,int f,int deep)
{
	fa[u]=f;
	siz[u]=1;
	dep[u]=deep;
	int maxson=-1;
	for(register int i=0;i<edge[u].size();i++)
	{
		int v=edge[u][i];
		if(v==f)continue;
		dfs1(v,u,deep+1);
		siz[u]+=siz[v];
		if(siz[v]>maxson)
		{
			maxson=siz[v];
			son[u]=v;
		}
	}
}
void dfs2(int u,int topf)
{
	top[u]=topf;
	id[u]=++cnt;
	if(!son[u])return;
	dfs2(son[u],topf);
	for(register int i=0;i<edge[u].size();i++)
	{
		int v=edge[u][i];
		if(v==fa[u]||v==son[u])continue;
		dfs2(v,v);
	}
}
inline void push_up(int u)
{
	tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
inline void push_down(int u)
{
	tr[u<<1].add+=tr[u].add;
	tr[u<<1|1].add+=tr[u].add;
	tr[u<<1].sum+=(tr[u<<1].r-tr[u<<1].l+1)*tr[u].add;
	tr[u<<1|1].sum+=(tr[u<<1|1].r-tr[u<<1|1].l+1)*tr[u].add;
	tr[u].add=0;
}
void build(int u,int l,int r)
{
	tr[u].l=l;
	tr[u].r=r;
	tr[u].add=0;
	if(l==r)
	{
		tr[u].sum=0;
		return;
	}
	int mid=l+r>>1;
	build(u<<1,l,mid);
	build(u<<1|1,mid+1,r);
	push_up(u);
}
void modify(int u,int l,int r,int v)
{
	if(tr[u].l>=l&&tr[u].r<=r)
	{
		tr[u].add+=v;
		tr[u].sum+=(tr[u].r-tr[u].l+1)*v;
	}
	else
	{
		push_down(u);
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid)modify(u<<1,l,r,v);
		if(r>mid)modify(u<<1|1,l,r,v);
		push_up(u); 
	}
}
int query(int u,int l,int r)
{
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u].sum;
	else
	{
		push_down(u);
		int ans=0;
		int mid=tr[u].l+tr[u].r>>1;
		if(l<=mid)ans+=query(u<<1,l,r);
		if(r>mid)ans+=query(u<<1|1,l,r);
		return ans;
	}
}
inline int R_change(int x,int y,int z)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		modify(1,id[top[x]],id[x],z);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	modify(1,id[x],id[y],z);
}
int main(void)
{
	n=read();
	for(register int i=1;i<=n;i++)
		a[i]=read();
	for(register int i=1;i<=n-1;i++)
	{
		int x,y;
		x=read();
		y=read();
		edge[x].push_back(y);
		edge[y].push_back(x);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	build(1,1,n);
	for(register int i=2;i<=n;i++)
	{
		int l=a[i-1];
		int r=a[i];
		R_change(l,r,1);
		modify(1,id[r],id[r],-1);
	}
	for(register int i=1;i<=n;i++)
		write(query(1,id[i],id[i])),puts("");
	return 0;
}
2022/8/17 10:57
加载中...