bzoj 3631: [JLOI2014]松鼠的新家

Description

松鼠的新家是一棵树，前几天刚刚装修了新家，新家有n个房间，并且有n-1根树枝连接，每个房间都可以相互到达，且俩个房间之间的路线都是唯一的。天哪，他居然真的住在“树”上。松鼠想邀请小熊维尼前来参观，并且还指定一份参观指南，他希望维尼能够按照他的指南顺序，先去a1，再去a2，……，最后到an，去参观新家。

可是这样会导致维尼重复走很多房间，懒惰的维尼不听地推辞。可是松鼠告诉他，每走到一个房间，他就可以从房间拿一块糖果吃。维尼是个馋家伙，立马就答应了。

现在松鼠希望知道为了保证维尼有糖果吃，他需要在每一个房间各放至少多少个糖果。因为松鼠参观指南上的最后一个房间an是餐厅，餐厅里他准备了丰盛的大餐，所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。

Input

第一行一个整数n，表示房间个数

第二行n个整数，依次描述a1-an

接下来n-1行，每行两个整数x，y，表示标号x和y的两个房间之间有树枝相连。

Output

一共n行，第i行输出标号为i的房间至少需要放多少个糖果，才能让维尼有糖果吃。

Sample Input

5
1 4 5 3 2
1 2
2 4
2 3
4 5

Sample Output

1
2
1
2
1

HINT

2<= n <=300000

树链剖分的题目在win下跑总是爆栈导致没法debug..比较怨念。只能换到lemon下加编译命令测。

这题可以先树链剖分，然后路径区间加即可了。

#include<string>
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
struct line
{
     int s,t;
     int next;
}a[600001];
int edge;
int head[300001];
inline void add(int s,int t)
{
	 a[edge].next=head[s];
     head[s]=edge;
     a[edge].s=s;
     a[edge].t=t;
}
struct tree
{
     int l,r;
     int s;
     int tag;
}tr[1200001];
inline int max(int x,int y)
{
     if(x>y)
          return x;
     return y;
}
inline void up(int p)
{
     tr[p].s=tr[p*2].s+tr[p*2+1].s;
}
inline void down(int p)
{
	 int l=tr[p].r-tr[p].l+1;
	 tr[p*2].s+=(l-l/2)*tr[p].tag;
	 tr[p*2+1].s+=l/2*tr[p].tag;
     tr[p*2].tag+=tr[p].tag;
     tr[p*2+1].tag+=tr[p].tag;
     tr[p].tag=0;
}
inline void build(int p,int l,int r)
{
     tr[p].l=l;
     tr[p].r=r;
     if(l!=r)
     {
          int mid=(l+r)/2;
          build(p*2,l,mid);
          build(p*2+1,mid+1,r);
     }
}
inline void change(int p,int l,int r,int x)//单点修改 l==r 故未加tag操作 
{
     if(l<=tr[p].l&&tr[p].r<=r)
     {
          tr[p].s+=(tr[p].r-tr[p].l+1)*x;
          tr[p].tag+=x;
     }
     else
     {
     	  down(p);
	      int mid=(tr[p].l+tr[p].r)/2;
		  if(l<=mid)
		       change(p*2,l,r,x);
		  if(r>mid)
		       change(p*2+1,l,r,x);
		  up(p);
     }
}
inline int find(int p,int l,int r)
{
     if(l<=tr[p].l&&tr[p].r<=r)
          return tr[p].s;
     else
     {
     	  down(p);
	      int mid=(tr[p].l+tr[p].r)/2;
	      int ans=0;
		  if(l<=mid)
		       ans+=find(p*2,l,r);
		  if(r>mid)
		       ans+=find(p*2+1,l,r);
		  return ans;
     }
}
int dep[300001],size[300001],son[300001],fa[300001];
int top[300001],w[300001];
int lson[300001]/*最大节点位置*/,mson[300001]/*最大节点值*/;
int tot;
inline void dfs1(int d)
{
     int i;
     for(i=head[d];i!=0;i=a[i].next)
     {
     	  int t=a[i].t;
          if(t!=fa[d])
          {
          	   dep[t]=dep[d]+1;
               fa[t]=d;
               dfs1(t);
               son[d]+=son[t]+1;
               if(son[t]>=mson[d])
               {
                    mson[d]=son[t];
                    lson[d]=t;
               }
          }
     }
}
inline void dfs2(int d)
{
     int i;
     for(i=head[d];i!=0;i=a[i].next)
     {
          int t=a[i].t;
          if(t==lson[d])
          {
               top[t]=top[d];
               tot++;
               w[t]=tot;
               dfs2(t);
          }
     }
     for(i=head[d];i!=0;i=a[i].next)
     {
          int t=a[i].t;
          if(t!=fa[d]&&t!=lson[d])
          {
               top[t]=t;
               tot++;
               w[t]=tot;
               dfs2(t);
          }
     }
}
inline void addx(int s,int t)
{
	 int u=top[s],v=top[t];
     while(u!=v)
     {
          if(dep[u]>dep[v])
          {
          	   change(1,w[u],w[s],1);
               s=fa[top[s]];
          }
          else
          {
          	   change(1,w[v],w[t],1);
               t=fa[top[t]];
          }
          u=top[s];
          v=top[t];
     }
     if(w[s]<w[t])
          change(1,w[s],w[t],1);
     else
          change(1,w[t],w[s],1);
}
int val[300001];
int main()
{
//     freopen("home.in","r",stdin);
//     freopen("home.out","w",stdout);
     int n;
     scanf("%d",&n);
     int i,s,t;
     for(i=1;i<=n;i++)
          scanf("%d",&val[i]);
     for(i=1;i<=n-1;i++)
     {
          scanf("%d%d",&s,&t);
          edge++;
          add(s,t);
          edge++;
          add(t,s);
     }
     dep[1]=1;
     dfs1(1);
     top[1]=1;
     tot++;
     w[1]=tot;
     dfs2(1);
     build(1,1,n);
     for(i=1;i<=n-1;i++)
     {
          addx(val[i],val[i+1]);
          if(i!=1)
               change(1,w[val[i]],w[val[i]],-1);
     }
     change(1,w[val[n]],w[val[n]],-1);
     for(i=1;i<=n;i++)
          printf("%d\n",find(1,w[i],w[i]));
     return 0;
}