Codeforces Div.2 682C Alyona and the Tree
問題要約
ある頂点から部分木の頂点までの距離をとする.
すべての, に対して, を満たさないようにするために、(木の)葉を取っていく.
取った数を調べる
やること
1. node 1 からの距離をDFSで調べる
2. だったら数え上げ
2. だったらそれをカウントせずにreturn
3. print(ノードの数 - 数え上げた数)
ここで, 負の辺があることに注意する。
distが負はありえない.
となることに注意する
ソースコード
#include <iostream> #include <queue> #include <map> #include <list> #include <vector> #include <string> #include <limits> #include <cassert> #include <fstream> #include <cstring> #include <bitset> #include <iomanip> #include <algorithm> #include <functional> #include <cstdio> #include <ciso646> #include <array> #include <cmath> using namespace std; #define FOR(i,a,b) for (int i=(a);i<(b);i++) #define RFOR(i,a,b) for (int i=(b)-1;i>=(a);i--) #define REP(i,n) for (int i=0;i<(n);i++) #define RREP(i,n) for (int i=(n)-1;i>=0;i--) #define FOREACH(i, a) for(int i=0;i<a.size();++i) #define inf 0x3f3f3f3f #define CLEAR(a) a = decltype(a)() #define MP make_pair #define ALL(a) (a).begin(),(a).end() #define pii pair<int ,int> #define pcc pair<char,char> #define pic pair<int,char> #define pci pair<char,int> #define VS vector<string> #define VI vector<int> #define DEBUG(x) cout<<#x<<": "<<x<<endl #define pi 2*acos(0.0) #define INFILE() freopen("in.txt","r",stdin) #define OUTFILE() freopen("out.txt","w",stdout) #define ll long long #define ull unsigned long long #define eps 1e-14 struct Edge { int to; int c; Edge(int to_, int c_):to(to_), c(c_){} }; int N; vector<int> A; vector<vector<Edge> > v; vector<bool> check; int dfs(int node, int dis) { if (check[node]) return 0; check[node] = true; if (dis > A[node]) return 0; int res = 1; for (auto &e : v[node]) { res += dfs(e.to, max(dis + e.c, 0)); } return res; } signed main(void) { cin >> N; A.resize(N); v.resize(N); check.resize(N); for (auto &a : A) cin >> a; for (int i = 1; i < N; ++i) { int p, c; cin >> p >> c; --p; v[i].emplace_back(p, c); v[p].emplace_back(i, c); } cout << N - dfs(0, 0) << endl; return 0; }