欢迎来到 xht37 的博客

CF1153D Serval and Rooted Tree

题目地址:CF1153D Serval and Rooted Tree

挺好玩儿也挺考思维的一道题

思路:树形DP+贪心

数组 \(d\) 维护这样一个值:

对于一个节点 \(x\) ,它的值最大可以为\(x\) 为根的子树中叶子节点的数值中排名第 \(d_x\) 大的数值

感性的理解就是,假如这个节点下有 \(n\) 个叶子节点,儿这个节点的 \(d\) 值为 \(k\) ,那么这个节点最大可以是 \(n+1-k\)

假设现在在节点 \(x\)

如果 \(x\) 是叶子节点,那么 \(d_x = 1\)

如果 \(x\)\(min\) 节点,那么 \(d_x = \sum_{y \in son_x}\ d_y\)

如果 \(x\)\(max\) 节点,那么 \(d_x = min_{y \in son_x}\ d_y\)

那么显然 \(ans = cnt + 1 - d_1\) ,其中 \(cnt\) 为叶子节点个数。

#include <bits/stdc++.h>
using namespace std;
const int N = 3e5 + 6;
int n, a[N], d[N], cnt;
vector<int> e[N];

void dfs(int x) {
    if (!e[x].size()) {
        d[x] = 1;
        ++cnt;
        return;
    }
    d[x] = a[x] ? n : 0;
    for (unsigned int i = 0; i < e[x].size(); i++) {
        int y = e[x][i];
        dfs(y);
        if (a[x]) d[x] = min(d[x], d[y]);
        else d[x] += d[y];
    }
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    for (int i = 2, f; i <= n; i++) {
        scanf("%d", &f);
        e[f].push_back(i);
    }
    dfs(1);
    cout << cnt + 1 - d[1] << endl;
    return 0;
}
posted @ 2019-04-14 16:33 xht37 阅读(...) 评论(...) 编辑 收藏