题目意思:
给定一颗root为1的树,节点最多5e5个,每个节点有一个英文字母,要求在指定子树根节点x,和指定深度y,求该子树y深度所有节点的字母能不能组成回文串。
解法1:
利用dfs序,为所有点标上进出两个标记,既可以代表一个区间(dfs序很重要的一个应用),顺便将每个深度下所有的单个字母出现的标号统计到相应的vector中后,
在计算答案时进行26次二分上下界求这个子树区间分别有多少a -- z即可。
解法2:
离线做法,先将所有问题储存,直接dfs一起求解。
用数组now 代表dfs到当前解,now[ i ] 代表所有已经走过的子树深度为i点a -- z个数的xor和。 所以dfs每个节点时先保存一下前面的解,然后当递归完该子树后,再去掉即可。
代码为第二种解法:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define rep1(i,x,y) for(int i=(int)x; i<=(int)y;i++)
const int N = 505050;
vector<pii> ans[N];
vector<int> G[N];
int n,m;
char str[N];
int now[N] , res[N];
void dfs(int u ,int de){
rep(i ,ans[u].size()){
int id = ans[u][i].first;
int dep = ans[u][i].second;
res[id]^=now[dep];
}
rep(i , G[u].size()){
dfs(G[u][i] ,de + 1);
}
now[de]^=(1<<(str[u] - 'a'));
rep(i ,ans[u].size()){
int id = ans[u][i].first;
int dep = ans[u][i].second;
res[id]^=now[dep];
}
}
int main()
{
scanf("%d %d",&n,&m);
rep1(i ,1 ,n) G[i].clear();
rep1(i , 2, n)
{
int x ;
scanf("%d",&x);
G[x].push_back(i);
}
scanf("%s",str + 1);
for(int i =1 ; i<=m; i++)
{
int x , dep;
scanf("%d %d",&x,&dep);
ans[x].push_back(pii(i , dep));
}
dfs(1 , 1);
rep1(i , 1 , m){
int hav = 0;
rep(j , 26) if((res[i]>>j) & 1)hav++;
printf("%s\n",hav<=1 ? "Yes" :"No");
}
return 0;
}