#include <cstdio> #include <queue> #include <vector> using namespace std; #define FOR(i,a,b) for(int i=(a);i<(b);++i) #define REP(i,n) FOR(i,0,n) #define VAR(v,w) __typeof(w) v=(w) #define FORE(it,c) for(VAR(it,(c).begin());it!=(c).end();++it) #define PB push_back #define INT(x) int x; scanf("%d", &x) #define LLG(x) LL x; scanf("%lld", &x) typedef long long LL; typedef vector<int> VI; typedef vector<VI> VVI; LL r[100000], c[200000]; int t[200000]; VVI g; bool u[100000]; queue<int>* qq; void go(int v, int p, LL b) { if (b < 0) return; if (!u[v]) { u[v] = 1; qq->push(v); } FORE(it,g[v]) { int w = t[*it]; if (w == p) continue; go(w, v, b - c[*it]); } } int main() { INT(n); REP(i,n) scanf("%lld", &r[i]); g.resize(n); int m = 0; REP(i,n-1) { INT(a); INT(b); LLG(cc); --a; --b; t[m] = b; c[m] = cc; g[a].PB(m++); t[m] = a; c[m] = cc; g[b].PB(m++); } REP(i,n) { if (i) printf(" "); REP(j,n) u[j] = 0; queue<int> q; qq = &q; u[i] = 1; q.push(i); while (!q.empty()) { int v = q.front(); q.pop(); go(v, -1, r[v]); } int res = 0; REP(j,n) if (u[j]) ++res; printf("%d", res); } printf("\n"); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | #include <cstdio> #include <queue> #include <vector> using namespace std; #define FOR(i,a,b) for(int i=(a);i<(b);++i) #define REP(i,n) FOR(i,0,n) #define VAR(v,w) __typeof(w) v=(w) #define FORE(it,c) for(VAR(it,(c).begin());it!=(c).end();++it) #define PB push_back #define INT(x) int x; scanf("%d", &x) #define LLG(x) LL x; scanf("%lld", &x) typedef long long LL; typedef vector<int> VI; typedef vector<VI> VVI; LL r[100000], c[200000]; int t[200000]; VVI g; bool u[100000]; queue<int>* qq; void go(int v, int p, LL b) { if (b < 0) return; if (!u[v]) { u[v] = 1; qq->push(v); } FORE(it,g[v]) { int w = t[*it]; if (w == p) continue; go(w, v, b - c[*it]); } } int main() { INT(n); REP(i,n) scanf("%lld", &r[i]); g.resize(n); int m = 0; REP(i,n-1) { INT(a); INT(b); LLG(cc); --a; --b; t[m] = b; c[m] = cc; g[a].PB(m++); t[m] = a; c[m] = cc; g[b].PB(m++); } REP(i,n) { if (i) printf(" "); REP(j,n) u[j] = 0; queue<int> q; qq = &q; u[i] = 1; q.push(i); while (!q.empty()) { int v = q.front(); q.pop(); go(v, -1, r[v]); } int res = 0; REP(j,n) if (u[j]) ++res; printf("%d", res); } printf("\n"); } |