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

Source

原题链接

说实话原题链接这个东西，，有时候记得我才加一下。。QAQ

这题对样例的理解花了我一定时间，，但是发现看题目描述会更加好懂。

大意不说了！

首先这题我觉得可以LCA+树上差分搞出来的。

但是我还是选树链剖分吧。

这题比较简单，维护都不用线段树。。（区间修改，单点查询）

维护一个数组s，x~y都加1时，s[x]++，s[y+1]--就好了（很好的一个思想）

接着对s作前缀和处理就行。注意一下仍然是树上id的位置。

明天要去上学。。本来觉着10分钟可以秒了的。。

结果竟然！快读错了好久没发现！！

花了足足20分钟啊。。。。。

再也不手打快读了

代码还是比较轻松的。。

其实跑起来速度还不错。（BZOJ上只有850+ms）

#include<bits/stdc++.h>using namespace std;int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0' || ch>'9'){if (ch=='-') f=-1;ch=getchar();}    while (ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}const int N=300005;int n,Ecnt,cnt;int a[N],s[N];struct Edge{int next,to;}E[N<<1]; int head[N];struct Tree{int pre,sz,son,top,tid,deep;}tree[N];void add(int u,int v){E[++Ecnt].next=head[u];E[Ecnt].to=v;head[u]=Ecnt;}void build(int x,int pre,int depth){tree[x].pre=pre;tree[x].sz=1;tree[x].son=0;tree[x].deep=depth;int maxx=0;for (int i=head[x];i;i=E[i].next){int j=E[i].to;if (j!=pre){build(j,x,depth+1);tree[x].sz+=tree[j].sz;if (maxx<tree[j].sz){maxx=tree[j].sz;tree[x].son=j;}}}}void getlink(int x,int ancestor){tree[x].top=ancestor;tree[x].tid=++cnt;if (tree[x].son) getlink(tree[x].son,ancestor);for (int i=head[x];i;i=E[i].next){int j=E[i].to;if (j!=tree[x].son && j!=tree[x].pre) getlink(j,j);}}void update(int x,int y){s[x]++; s[y+1]--;}void work(int u,int v){while (tree[u].top!=tree[v].top){if (tree[tree[u].top].deep<tree[tree[v].top].deep) swap(u,v);update(tree[tree[u].top].tid,tree[u].tid);u=tree[tree[u].top].pre;}if (tree[u].deep>tree[v].deep) swap(u,v);    update(tree[u].tid,tree[v].tid);}int main(){n=read();for (int i=1;i<=n;i++) a[i]=read();int u,v; Ecnt=0;for (int i=1;i<n;i++){u=read(),v=read();add(u,v); add(v,u);}build(1,0,0);cnt=0; getlink(1,1);for (int i=1;i<n;i++)work(a[i],a[i+1]);s[0]=0;for (int i=1;i<=n;i++)s[i]+=s[i-1];s[tree[a[1]].tid]++;for(int i=1;i<=n;i++)printf("%d\n",s[tree[i].tid]-1);return 0;}