bzoj 3631 [JLOI2014]松鼠的新家

题目

传送门

题解

看到某Yuta的树总结觉得LCA+树上差分挺好写而且差分没写过准备水一发结果水了一上午qwq。因为第一次写树上差分和树剖求LCA，结果树剖版子某处打错了。。（而且差分点和差分边还是有一定区别的）。

---

这题可以用树剖水部分分，不过n有30w，log方就会被卡了。所以利用差分思想把区间修改转化为点修改。给每个点一个tag表示新增值（貌似怪怪的）。如果有条路径（i，j）需要标记那么进行如下操作

tag（i）++；tag（j）++；tag（LCA（i，j））- -；tag（father（LCA（i，j）））- -；

这样能够达到标记经过的点的效果，请自行脑补如果是经过的边如何标记。
那么就把路径拆分就好了，但是这道题非起始端点会被多算一次（在衔接处），所以统计完后减回来。

---

题很水，人很蠢。

代码

//QWsin
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=600000+10;
const int maxm=1200000+10;
inline int read()
{
    int ret=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-'0';
    return ret;
}

int to[maxn];

int first[maxn],next[maxm],ecnt;
struct Edge{int u,v;Edge(int u=0,int v=0):u(u),v(v){}}e[maxm];
inline void add_edge(int u,int v)
{
    next[ecnt]=first[u];first[u]=ecnt;e[ecnt++]=Edge(u,v);
    next[ecnt]=first[v];first[v]=ecnt;e[ecnt++]=Edge(v,u);
}

int top[maxn],hv[maxn],sz[maxn],p[maxn],dep[maxn];
int tag[maxn];  

int dfs1(int u,int fa,int deep=1)
{
    p[u]=fa;dep[u]=deep;sz[u]=1;int msz=0;
    for(int i=first[u];i!=-1;i=next[i]) if(e[i].v!=fa){
        int lch=dfs1(e[i].v,u,deep+1);
        if(lch > msz) msz=lch,hv[u]=e[i].v;
        sz[u]+=lch;
    }
    return sz[u];
}

void dfs2(int u,int Top)
{
    top[u]=Top;
    if(sz[u]==1) return ;
    dfs2(hv[u],Top);
    for(int i=first[u];i!=-1;i=next[i])
        if(e[i].v!=p[u]&&e[i].v!=hv[u]) dfs2(e[i].v,e[i].v);
}

int LCA(int a,int b)
{
    for(;top[a]!=top[b];b=p[top[b]])
        if(dep[top[a]] > dep[top[b]]) swap(a,b);
    return dep[a]<dep[b]?a:b;
}

int ans[maxn],sum,n;
int dfs3(int cur)
{
    int sum=tag[cur];
    for(int i=first[cur];i!=-1;i=next[i]) 
        if(e[i].v!=p[cur]) sum+=dfs3(e[i].v);
    if(to[1]!=cur) ans[cur]=sum-1;
    else ans[cur]=sum; return sum;
}

int main()
{
    n=read();memset(first,-1,sizeof(first));
    for(int i=1;i<=n;i++) to[i]=read();
    for(int i=1;i<n;i++) add_edge(read(),read());

    int root=1;
    dfs1(root,root);dfs2(root,root);

    tag[to[1]]++;tag[to[2]]++;
    int lca=LCA(to[1],to[2]);
    tag[lca]--;if(lca!=root) tag[p[lca]]--;

    for(int i=2;i<n;i++)
    {
        int lca=LCA(to[i],to[i+1]);
        tag[to[i]]++;tag[to[i+1]]++;
        tag[lca]--;if(lca!=root) tag[p[lca]]--;
    }

    dfs3(root);
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    return 0;
}