dfs序 二進制優化 Codeforces Round #316 (Div. 2)D. Tree Requests

D. Tree Requests
time limit per test2 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
Roman planted a tree consisting of n vertices. Each vertex contains a lowercase English letter. Vertex 1 is the root of the tree, each of the n - 1 remaining vertices has a parent in the tree. Vertex is connected with its parent by an edge. The parent of vertex i is vertex pi, the parent index is always less than the index of the vertex (i.e., pi < i).node

The depth of the vertex is the number of nodes on the path from the root to v along the edges. In particular, the depth of the root is equal to 1.ios

We say that vertex u is in the subtree of vertex v, if we can get from u to v, moving from the vertex to the parent. In particular, vertex v is in its subtree.c++

Roma gives you m queries, the i-th of which consists of two numbers vi, hi. Let’s consider the vertices in the subtree vi located at depth hi. Determine whether you can use the letters written at these vertices to make a string that is a palindrome. The letters that are written in the vertexes, can be rearranged in any order to make a palindrome, but all letters should be used.數組

Input
The first line contains two integers n, m (1 ≤ n, m ≤ 500 000) — the number of nodes in the tree and queries, respectively.app

The following line contains n - 1 integers p2, p3, …, pn — the parents of vertices from the second to the n-th (1 ≤ pi < i).less

The next line contains n lowercase English letters, the i-th of these letters is written on vertex i.ide

Next m lines describe the queries, the i-th line contains two numbers vi, hi (1 ≤ vi, hi ≤ n) — the vertex and the depth that appear in the i-th query.ui

Output
Print m lines. In the i-th line print 「Yes」 (without the quotes), if in the i-th query you can make a palindrome from the letters written on the vertices, otherwise print 「No」 (without the quotes).spa

Examples
inputCopy
6 5
1 1 1 3 3
zacccd
1 1
3 3
4 1
6 1
1 2
outputCopy
Yes
No
Yes
Yes
Yes
Note
String s is a palindrome if reads the same from left to right and from right to left. In particular, an empty string is a palindrome..net

Clarification for the sample test.

In the first query there exists only a vertex 1 satisfying all the conditions, we can form a palindrome 「z」.

In the second query vertices 5 and 6 satisfy condititions, they contain letters 「с」 and 「d」 respectively. It is impossible to form a palindrome of them.

In the third query there exist no vertices at depth 1 and in subtree of 4. We may form an empty palindrome.

In the fourth query there exist no vertices in subtree of 6 at depth 1. We may form an empty palindrome.

In the fifth query there vertices 2, 3 and 4 satisfying all conditions above, they contain letters 「a」, 「c」 and 「c」. We may form a palindrome 「cac」.

dfs序學自:https://blog.csdn.net/u012061345/article/details/54023285
簡單來講,就是記錄dfs中每個節點入棧和出棧的時間,若是a節點入棧時間在b節點入棧出棧時間之間的話,a即爲b子樹上的節點。
用一個二維數組存儲每一個深度全部的節點的到達時間,並用二進制數記錄這個節點是什麼字母,搜索時先鎖定深度,在從中找到v的子節點,將這些節點進行異貨,一串字符能組成迴文串當且僅當最多有一個字母出現次數爲奇數,所以判斷異貨後的數中1的個數便可。

/* *********************************************** Author :ACagain Created Time :2018/4/10 19:53:20 File Name :4_10.cpp ************************************************ */

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <vector>
using namespace std;

#define lson o<<1,l,m
#define rson o<<1|1,m+1,r
#define pii pair<int,int>
#define mp make_pair
#define ll long long
#define INF 0x3f3f3f3f
const int maxn=5*1e5+2;
int n,m,t;
char c[maxn];
vector<int> son[maxn];
vector<int> d[maxn];     //d[i][j] i深度第j個節點到達時間
int l[maxn],r[maxn];     //v入棧出棧時間
vector<int> hsh[maxn];   //對應字母
void dfs(int x,int depth)
{
    l[x]=++t;
    int k=son[x].size();
    int tmp=1<<(int(c[x]-'a'));
    d[depth].push_back(t);
    int tt=hsh[depth].size();
    if(tt==0)
      hsh[depth].push_back(tmp);
    else
      hsh[depth].push_back(tmp^hsh[depth][tt-1]);
    for(int i=0;i<k;i++)
      dfs(son[x][i],depth+1);
    r[x]=t;
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    scanf("%d%d",&n,&m);
    int tmp;
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&tmp);
        son[tmp].push_back(i);
    }
    for(int i=1;i<=n;i++)
      scanf(" %c",&c[i]);
    t=-1;
    dfs(1,1);
    int v,h,num;
    while(m--)
    {
        scanf("%d%d",&v,&h);
        num=0;
        if(d[h].size()==0)
        {
          printf("%s\n","Yes");
          continue;
        }
        int L=lower_bound(d[h].begin(),d[h].end(),l[v])-d[h].begin()-1;
        int R=upper_bound(d[h].begin(),d[h].end(),r[v])-d[h].begin()-1;
        int tmp=hsh[h][R];
        if(R<0)
        {
            printf("%s\n","Yes");
            continue;
        }
        if(L>=0)
          tmp^=hsh[h][L];
        while(tmp)
        {
            if(tmp&1==1)
              num++;
            tmp>>=1;
        }
        if(num>1)
        printf("%s\n","No");
        else
        printf("%s\n","Yes");
    }
    return 0;
}
相關文章
相關標籤/搜索