*Codeforces Round #316 (Div. 2)- D. Tree Requests (dfs+二分)

题目:

http://codeforces.com/contest/570/problem/D

题意:

n个点的一棵树,每个节点都有自己的字母,m个询问,问u节点的子树中高度为h的点是否可以组成回文序列。

思路:

dfs,求出每个点的dfs序,以及记录每个深度的节点的字母的序。

字母数量的奇数个不超过1,则可以组成回文序列。二分求出询问节点对应的区间,判断是否可以。

AC.

#include 
#include 
#include 
#include 
#include 

using namespace std;
const int maxn = 5e5+5;
int n, m;
char let[maxn];

int tol, head[maxn];
struct Edge{
    int to, next;
}edge[maxn];
void addedge(int u, int v)
{
    edge[tol].to = v;
    edge[tol].next = head[u];
    head[u] = tol++;
}

vector vec[maxn][30];
int in[maxn], ou[maxn], tot;
void dfs(int u, int d)
{
    in[u] = ++tot;
    vec[d][let[u]-'a'].push_back(tot);
    for(int i = head[u]; ~i; i = edge[i].next) {
        int v = edge[i].to;
        dfs(v, d+1);
    }
    ou[u] = ++tot;
}

void init()
{
    tol = 0;
    tot = 0;
    memset(head, -1, sizeof(head));
    for(int i = 0; i <= n; ++i) {
        for(int j = 0; j <= 26; ++j) {
            vec[i][j].clear();
        }
    }
}

int main()
{
    //freopen("in", "r", stdin);
    while(~scanf("%d%d", &n, &m)) {
        int u, v;
        init();
        for(int i = 2; i <= n; ++i) {
            scanf("%d", &u);
            addedge(u, i);
        }
        scanf("%s", let+1);
        dfs(1, 1);

        while(m--) {
            scanf("%d %d", &u, &v);

            int res = 0, ok = 0;
            for(int i = 0; i < 26; ++i) {
                res = lower_bound(vec[v][i].begin(), vec[v][i].end(), ou[u]) -
                      lower_bound(vec[v][i].begin(), vec[v][i].end(), in[u]-1);
                if(res % 2) ok++;
                if(ok > 1) break;
            }
            if(ok > 1) printf("No\n");
            else printf("Yes\n");
        }
    }
    return 0;
}




你可能感兴趣的:(二分)