CodeForces - 570D Tree Requests (树上启发式合并)

树上启发式合并,题目需要求的是每棵子树独立的一个答案,此时考虑树上启发式合并。
判断是否可以组成回文只需要满足小于等于一个字母出现奇数次。
在做树上启发式合并的时候合并操作需要根据后续提问来标记答案。因为没办法在处理的时候记下所有答案。所以我们需要询问的一个数组。以顶点为集合标识,在里面遍历,找到一个深度就计算是否可行,然后这个顶点下的这个深度的ans就有了,我们给这个答案一个编号以便于后续输出。

//
// Created by SANZONG on 2021/7/6.
//

#include "bits/stdc++.h"

using namespace std;
const int maxn = 6e5;
struct node {
    int next, to;
} a[maxn<<1];
struct Query {
    int no, h; //no为输入编号
};
vector<Query> p[maxn];  //存储输入的结构
int cnt = 0;
int head[maxn << 1];
char s[maxn];
void add(int u, int v) {
    a[++cnt].to = v;
    a[cnt].next = head[u];
    head[u] = cnt;
}
int siz[maxn];
int son[maxn];
int dep[maxn];
void dfsFir(int u, int fa, int d) {
    siz[u] = 1;
    dep[u] = d;
    for (int i = head[u]; i; i = a[i].next) {
        int v = a[i].to;
        if (v == fa) continue;
        dfsFir(v, u, d + 1);
        siz[u] += siz[v];
        if (siz[v] > siz[son[u]]) {
            son[u] = v;
        }
    }
}
int flag;
int h[maxn][26];
void count(int u, int fa, int val) {
    h[dep[u]][s[u] - 'a'] += val;
    for (int i = head[u]; i; i = a[i].next) {
        int v = a[i].to;
        if (v == fa || v == flag) continue;
        count(v, u, val);
    }
}
int ans[maxn];
void dfsSec(int u, int fa, bool keep) {
    for (int i = head[u]; i; i = a[i].next) {
        int v = a[i].to;
        if (v == fa || v == son[u]) continue;
        dfsSec(v, u, false);
    }
    if (son[u]) {
        dfsSec(son[u], u, true);
        flag = son[u];
    }
    count(u, fa, 1);
    flag = 0;
    for (Query q : p[u]) {
        int t = 0, ff = 0;
        for (int i = 0; i <= 25; ++i) {
            if (h[q.h][i] % 2) t++;
        }
        if (t >= 2) {
            ans[q.no] = 0;
            ff = 1;
        }
        if (!ff) ans[q.no] = 1;
    }
    if (!keep) {
        count(u, fa, -1);
    }

}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    int n, q;
    cin >> n >> q;
    for (int i = 1; i <= n - 1; ++i) {
        int u;
        cin >> u;
        add(u, i + 1);
        add(i + 1, u);
    }
    cin >> (s + 1);
    for (int i = 1; i <= q; ++i) {
        int x, y;
        cin >> x >> y;
        p[x].push_back(Query{i, y});
    }
    dfsFir(1, 0, 1);
    dfsSec(1, 0, 0);
    for (int i = 1; i <= q; ++i) {
        cout << (ans[i] ? "Yes" : "No") << endl;
    }
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值