CodeForces 682C - Alyona and the Tree(dfs)
题意:给定一颗无向树(根节点编号为1),节点权值(1 ~ 10^9) < 由某个点至该节点边的权值(-10^9 ~ 10^9)之和 的节点称为sad点,求至少删除多少个节点才能使这棵树不存在sad点。
从根出发dfs即可,找到sad点则return,对遍历到的点计数,然后答案即为节点数 - 计数个数。
若边的和出现负值,则取0:
1、对于正在研究的节点来说,输入的权值必定为正(1 ~ 10^9),必定符合条件。
2、对于接下来的节点来说,相当于把这个让边权值和为负的点,当作祖宗,继续研究接下来的节点(0与某个边的权值相加)。
(对于sad点,即使它下边的分支为非sad点,那么删也要删这整棵子树)
需要用long long类型。
#include<cstdio> #include<cstring> #include<cctype> #include<cstdlib> #include<cmath> #include<iostream> #include<sstream> #include<iterator> #include<algorithm> #include<string> #include<vector> #include<set> #include<map> #include<deque> #include<queue> #include<stack> #include<list> typedef long long ll; typedef unsigned long long llu; const int MAXN = 100 + 10; const int MAXT = 100000 + 10; const int INF = 0x7f7f7f7f; const double pi = acos(-1.0); const double EPS = 1e-6; using namespace std; int n; ll a[MAXT], ans; map<int, ll> mp[MAXT]; void dfs(int lur, ll num){ if(num > a[lur]) return; ++ans; int len = mp[lur].size(); for(map<int, ll>::iterator i = mp[lur].begin(); i != mp[lur].end(); ++i) dfs((*i).first, max((*i).second + num, 0ll)); } int main(){ scanf("%d", &n); int tmp1; ll tmp2; ans = 0ll; for(int i = 1; i <= n; ++i) scanf("%I64d", a + i); for(int i = 1; i < n; ++i){ scanf("%d%I64d", &tmp1, &tmp2); mp[tmp1][i + 1] = tmp2; } dfs(1, 0ll); printf("%I64d\n", ll(n) - ans); return 0; }