Codeforces Div.2 682C Alyona and the Tree

問題

codeforces.com

問題要約

ある頂点から部分木の頂点までの距離をとする.
すべての, に対して, を満たさないようにするために、(木の)葉を取っていく.
取った数を調べる

やること

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;
}