[CF570D]Tree Requests

570D:Tree Requests

题意简述

给出一棵$n$个节点的树，每个节点上有一个小写字母。
下面有$m$个询问，每个询问形如$x$的子树中深度为$h$的节点上的字符，能否构成一个回文串。

数据范围

$1 \leq n,m \leq 5 \ast 10^5$

思路

树链剖分。
启发式合并，轻链向重链合并。
每个节点最多合并$logn$次。
维护每个字母的奇偶性，每个深度奇数字母的个数。
能构成回文串当且仅当奇数字母 <2 <script type="math/tex" id="MathJax-Element-7"><2</script>
时间复杂度$O(nlogn)$

代码

#include<cstdio>
#include<cstring>
using namespace std;
struct edge{
    int s,t,next;
}e[1000010];
int head[500010],cnt;
void addedge(int s,int t)
{
    e[cnt].s=s;e[cnt].t=t;e[cnt].next=head[s];head[s]=cnt++;
    e[cnt].s=t;e[cnt].t=s;e[cnt].next=head[t];head[t]=cnt++;
}
struct query{
    int pos,h,next;
}q[500010];
int head_q[500010],cnt_q;
void add_q(int _pos,int _h)
{
    q[cnt_q].pos=_pos;q[cnt_q].h=_h;q[cnt_q].next=head_q[_pos];head_q[_pos]=cnt_q++;
}
int n,m,u,v;
int size[500010],son[500010],deep[500010],tot[500010];
bool ans[500010],ski[500010],num[500010][26];
char st[500010];
void dfs(int node,int lastfa,int de)
{
    size[node]=1;
    deep[node]=de;
    for (int i=head[node];i!=-1;i=e[i].next)
        if (e[i].t!=lastfa)
        {
            dfs(e[i].t,node,de+1);
            size[node]+=size[e[i].t];
            if (size[e[i].t]>size[son[node]])
                son[node]=e[i].t;
        }
}
void modify(int node,int lastfa)
{
    num[deep[node]][st[node]-'a']^=1;
    if (num[deep[node]][st[node]-'a'])
        tot[deep[node]]++;
    else
        tot[deep[node]]--;
    for (int i=head[node];i!=-1;i=e[i].next)
        if (e[i].t!=lastfa&&!ski[e[i].t])
            modify(e[i].t,node);
}
void dfs2(int node,int lastfa,bool keep)
{
    for (int i=head[node];i!=-1;i=e[i].next)    
        if (e[i].t!=lastfa&&e[i].t!=son[node])
            dfs2(e[i].t,node,0);
    if (son[node])
    {
        dfs2(son[node],node,1);
        ski[son[node]]=1;
    }
    modify(node,lastfa);
    for (int i=head_q[node];i!=-1;i=q[i].next)
        if (tot[q[i].h]<=1)
            ans[i+1]=1;
    ski[son[node]]=0;
    if (!keep)
        modify(node,lastfa);
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(head,0xff,sizeof(head));
    cnt=0;
    for (int i=2;i<=n;i++)
    {
        scanf("%d",&u);
        addedge(i,u);
    }
    scanf("%s",st+1);
    memset(head_q,0xff,sizeof(head_q));
    cnt_q=0;
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&u,&v);
        add_q(u,v);
    }
    dfs(1,1,1);
    dfs2(1,1,1);
    for (int i=1;i<=m;i++)
        if (ans[i])
            printf("Yes\n");
        else
            printf("No\n");
    return 0;
}