【BZOJ-3631】松鼠的新家 树形DP？+ 倍增LCA + 打标记

时间：2023-03-09 14:38:47
【BZOJ-3631】松鼠的新家 树形DP？+ 倍增LCA + 打标记

3631: [JLOI2014]松鼠的新家

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1231  Solved: 620
[Submit][Status][Discuss]

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

Solution

一眼树链剖分裸题，但是懒得打.....这里是一种巧妙的方法: LCA后打标记,树形DP对标记进行上传统计答案

方法就是对于从a[i]走到a[i+1]这个路径,对a[i]和a[i+1]打上start标记,对LCA(a[i],a[i+1])和LCA的父节点打上end标记

把所有的标记上传,更新答案;

自己的理解是因为在向上更新的时候,a[i]和a[i+1]的标记更新到LCA抵消后会多出一个额外的标记,所以需要两个end标记

至于最后答案减一?因为出始节点是不算入的,但上传标记需要给初始节点打标记,所以得减一(这个还是很好想的)

Code

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<cmath>

using namespace std;

int read()

{

    int x=,f=; char ch=getchar();

    while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}

    while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}

    return x*f;

}

#define maxn 301000

int n,a[maxn];

struct EdgeNode{int next,to,from;}edge[maxn<<];

int head[maxn],cnt;

void add(int u,int v)

{

    cnt++;edge[cnt].next=head[u];head[u]=cnt;edge[cnt].to=v;edge[cnt].from=u;

}

void insert(int u,int v) {add(u,v); add(v,u);}

int deep[maxn],father[maxn][],ans[maxn],start[maxn],end[maxn];

void DFS(int x)

{

    for (int i=; i<=; i++)

        if (deep[x]>=(<<i)) father[x][i]=father[father[x][i-]][i-];

            else break;

    for (int i=head[x]; i; i=edge[i].next)

        if (edge[i].to!=father[x][])

            {

                deep[edge[i].to]=deep[x]+;

                father[edge[i].to][]=x;

                DFS(edge[i].to);

            }

}

int LCA(int x,int y)

{

    if (deep[x]<deep[y]) swap(x,y);

    int dd=deep[x]-deep[y];

    for (int i=; i<=; i++)

        if (dd&(<<i)) x=father[x][i];

    for (int i=; i>=; i--)

        if (father[x][i]!=father[y][i])

            x=father[x][i],y=father[y][i];

    if (x==y) return x; else return father[x][];

}

void UpTag(int x)

{

    ans[x]=start[x]-end[x];

    for (int i=head[x]; i; i=edge[i].next)

        if (edge[i].to!=father[x][])

            UpTag(edge[i].to),ans[x]+=ans[edge[i].to];

}

int main()

{

    n=read();

    for (int i=; i<=n; i++) a[i]=read();

    for (int u,v,i=; i<=n-; i++) u=read(),v=read(),insert(u,v);

    DFS(a[]);

    for (int lca,i=; i<=n-; i++)

        start[a[i]]++,start[a[i+]]++,lca=LCA(a[i],a[i+]),end[lca]++,end[father[lca][]]++;

    UpTag(a[]);

    for (int i=; i<=n; i++) ans[a[i]]--;

    for (int i=; i<=n; i++) printf("%d\n",ans[i]);

    return ;

}