abc309E 统计家族中已买保险的人数
题面:一个有n个成员的家族构成一棵树,1为树根。有m份保险,第i份保险为x[i]以及他的y[i]代以内的所有子孙购买。问总共有多少人有保险。
范围:1<=n,m<=3e5
分析:dfs遍历统计即可,y值需要向下传递,-1表示未买。
#include <bits/stdc++.h>
using i64 = long long;
const int N = 300005;
int n, m, f[N], ok[N];
std::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() {
std::cin >> n >> m;
for (int i = 2; i <= n; i++) {
int p;
std::cin >> p;
adj[p].push_back(i);
adj[i].push_back(p);
}
for (int i = 1; i <= n; i++) {
f[i] = -1;
}
for (int i = 1; i <= m; i++) {
int x, y;
std::cin >> x >> y;
f[x] = std::max(f[x], y);
}
dfs(1, 1, -1);
int ans = 0;
for (int i = 1; i <= n; i++) {
if (ok[i]) {
ans++;
}
}
std::cout << ans << "\n";
}
int main() {
std::cin.tie(0)->sync_with_stdio(0);
int t = 1;
while (t--) solve();
return 0;
}
浙公网安备 33010602011771号