abc309E 统计家族中已买保险的人数

题面:一个有n个成员的家族构成一棵树,1为树根。有m份保险,第i份保险为x[i]以及他的y[i]代以内的所有子孙购买。问总共有多少人有保险。
范围:1<=n,m<=3e5

思路:dfs遍历统计即可,y值需要向下传递,-1表示未买。

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define rep(i,a,b) for(int i=a; i<=b; i++)
#define per(i,a,b) for(int i=b; i>=a; i--)

const int N = 300005;
int n, m, f[N], ok[N];
vector<int> adj[N];
void dfs(int x, int p, int z) {
    f[x] = max(f[x], z);
    if (f[x] >= 0) ok[x] = 1;
    for (auto i : adj[x]) if (i != p) {
        dfs(i, x, f[x]-1);
    }
}
void solve() {
    cin >> n >> m;
    rep(i,2,n) {
        int p;
        cin >> p;
        adj[p].push_back(i);
        adj[i].push_back(p);
    }
    rep(i,1,n) f[i] = -1;
    rep(i,1,m) {
        int x, y;
        cin >> x >> y;
        f[x] = max(f[x], y);
    }
    dfs(1, 1, -1);
    int ans = 0;
    rep(i,1,n) if (ok[i]) ans++;
    cout << ans << "\n";
}

signed main() {
    cin.tie(0)->sync_with_stdio(0);
    int t = 1;
    while (t--) solve();
    return 0;
}
posted @ 2024-03-07 21:26  chenfy27  阅读(4)  评论(0编辑  收藏  举报