[Codeforces570D]Tree Requests（dsu on the tree）

题目描述

传送门
题意：一棵树，每一个点有一个字符，给出若干询问，每一次询问以某个点为根的子树中深度为x的点能否组成一个回文串。

题解

用dsu on the tree搞一下。。
如果可以组成一个回文串的话出现次数为奇数的字符至多有一个
记录一下深度为i的点中字符为j的个数，然后对于每一个点的询问搞一个链表查询就可以了
时间复杂度O(nlogn)

代码

#include<algorithm>#include<iostream>#include<cstring>#include<cstdio>#include<cmath>using namespace std;#define N 500005int n,m,Son;char s[N];int tot,point[N],nxt[N],v[N];int cq,pq[N],nxtq[N],vq[N],idq[N];int size[N],h[N],son[N],cnt[N][30];bool ans[N];void add(int x,int y){    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;}void addq(int x,int y,int z){    ++cq; nxtq[cq]=pq[x]; pq[x]=cq; vq[cq]=y; idq[cq]=z;}void getson(int x,int dep){    size[x]=1;h[x]=dep;    for (int i=point[x];i;i=nxt[i])    {        getson(v[i],dep+1);        size[x]+=size[v[i]];        if (size[v[i]]>size[son[x]]) son[x]=v[i];    }}void calc(int x,int opt){    cnt[h[x]][s[x]-'a']+=opt;    for (int i=point[x];i;i=nxt[i])        if (v[i]!=Son) calc(v[i],opt);}void dfs(int x,int k){    for (int i=point[x];i;i=nxt[i])        if (v[i]!=son[x]) dfs(v[i],0);    if (son[x]) dfs(son[x],1),Son=son[x];    calc(x,1);Son=0;    for (int i=pq[x];i;i=nxtq[i])    {        int H=vq[i],odd=0;        for (int j=0;j<26;++j)            if (cnt[H][j]&1) ++odd;        if (odd<=1) ans[idq[i]]=1;    }    if (!k) calc(x,-1);}int main(){    scanf("%d%d",&n,&m);    for (int i=2;i<=n;++i)    {        int fa;scanf("%d",&fa);        add(fa,i);    }    scanf("%s",s+1);    for (int i=1;i<=m;++i)    {        int x,y;scanf("%d%d",&x,&y);        addq(x,y,i);    }    getson(1,1);    dfs(1,0);    for (int i=1;i<=m;++i)        if (ans[i]) puts("Yes");        else puts("No");}