Codeforces 570D,dfs+位运算

题意:

给你一棵n个节点的树,每个节点上标了一个小写字母(‘a’~‘z’),现在定义,根节点为节点1,且它的深度为1,其他节点的深度随树的形状,依次确定。

现在有m个查询,查询方式:给你一个节点u和深度h,问以节点u为根节点的子树里面,深度为h(这里的h是从节点1开始计算的)的所有节点上面的字母,(可以对字母任意进行排列组合)能否构成一个回文串。


题解:

因为所有查询的h都是从节点1开始的,这个非常关键,当时自己就是读错题了,所以怎么想都不会做。

既然都是从节点1开始计算,则我们可以通过一个数组state[N],来表示高度为h的所有节点的状态。

拥有的字母,这里通过1<<27的二进制状态来存储。因为对某个高度上的字母可以随意排列,所以只需考虑每个字母的数量,成为回文串的前提,最多允许一个字母的数量是奇数,这里通过异或操作(^)来进行计算,若最后计算结果的二进制表示中,只有一个或零个1,则可以构成回文串。


对于所有查询,采用离线的方式,来进行计算。

关键:将回文串求解,转化为了某个字母的数量是否为奇数,技巧:把字母的状态通过1<<27的状态存下来,然后通过异或操作,判断最后的结果二进制表示方法有几个1,若有1个或0个,则为回文串,否则不是。

实现:通过深搜dfs来求解,在回溯的过程中,来进行更新子树,并求解查询结果。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<vector>
using namespace std;
#define N 500010
typedef long long LL;
int n,m;
char s[N];
vector<int> edge[N];
vector<pair<int,int> > query[N];
int ans[N],st[N];
void dfs(int u,int d)
{
    for(int i=0;i<query[u].size();i++)
    {
        pair<int,int> t=query[u][i];
        int query_id,query_dep;
        query_id=t.first;
        query_dep=t.second;
        ans[query_id]^=st[query_dep];
    }
    for(int i=0;i<edge[u].size();i++)
    {
        int next=edge[u][i];
        dfs(next,d+1);
    }
    st[d]^=1<<(s[u-1]-'a');
    for(int i=0;i<query[u].size();i++)
    {
        pair<int,int> t=query[u][i];
        int query_id,query_dep;
        query_id=t.first;
        query_dep=t.second;
        ans[query_id]^=st[query_dep];
    }
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(st,0,sizeof(st));
        memset(ans,0,sizeof(ans));
        for(int i=0;i<n;i++)
        {
            edge[i].clear();
            query[i].clear();
        }
        for(int i=0;i<n-1;i++)
        {
            int v;
            scanf("%d",&v);
            edge[v].push_back(i+2);
        }
        scanf("%s",s);
        for(int i=0;i<m;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            query[a].push_back(make_pair(i,b));
        }
        dfs(1,1);
        for(int i=0;i<m;i++)
            printf("%s\n",ans[i]&(ans[i]-1)?"No":"Yes");
    }
    return 0;
}
/*
6 5
1 1 1 3 3
zacccd
1 1
3 3
4 1
6 1
1 2


*/


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值