QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#258404#7578. Salty Fishddl_VS_pigeon#AC ✓813ms94524kbC++173.2kb2023-11-19 17:44:452023-11-19 17:44:45

Judging History

This is the latest submission verdict.

  • [2023-11-19 17:44:45]
  • Judged
  • Verdict: AC
  • Time: 813ms
  • Memory: 94524kb
  • [2023-11-19 17:44:45]
  • Submitted

answer

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const ll lnf = 1e18;

struct Segt {
	struct P {
		ll mx, tag;
		P() : mx(0), tag(0) {}
		void add(ll val) {
			mx += val, tag += val;
		}
	};
	int n;
	vector<P> t;
	Segt(int _n) : n(_n), t(n << 2 | 1) {}
	void add(int al, int ar, ll val) {
		add(al, ar, val, 1, 0, n - 1);
	}
	ll getmx(int ql, int qr) {
		return getmx(ql, qr, 1, 0, n - 1);
	}
	void print() {
		print(0, 1, 0, n - 1);
		cout << endl;
	}
	#define mid ((l + r) >> 1)
	#define ls (v << 1)
	#define rs (v << 1 | 1)
	void add(int al, int ar, ll val, int v, int l, int r) {
		if (ar < l || al > r) {
			return;
		}
		if (al <= l && ar >= r) {
			t[v].add(val);
		} else {
			add(al, ar, val, ls, l, mid);
			add(al, ar, val, rs, mid + 1, r);
			t[v].mx = max(t[ls].mx, t[rs].mx) + t[v].tag;
		}
	}
	ll getmx(int ql, int qr, int v, int l, int r) {
		if (qr < l || ql > r) {
			return -lnf;
		}
		if (ql <= l && qr >= r) {
			return t[v].mx;
		}
		return max(getmx(ql, qr, ls, l, mid), getmx(ql, qr, rs, mid + 1, r)) + t[v].tag;
	}
	void print(ll d, int v, int l, int r) {
		d += t[v].tag;
		if (l == r) {
			cout << d << ' ';
		} else {
			print(d, ls, l, mid);
			print(d, rs, mid + 1, r);
		}
	}
	#undef rs
	#undef ls
	#undef mid
};

struct Tree {
	int n;
	vector<vector<int>> son;
	vector<int> a;
	vector<vector<pair<int, int>>> p;
	vector<int> wson, top, dep;
	Tree(int _n, int m) : n(_n), son(n + 1), a(n + 1), p(n + 1), wson(n + 1), top(n + 1), dep(n + 1) {
		for (int i = 2; i <= n; i++) {
			int f;
			cin >> f;
			son[f].emplace_back(i);
		}
		for (int i = 1; i <= n; i++) {
			cin >> a[i];
		}
		for (int i = 1; i <= m; i++) {
			int x, k, c;
			cin >> x >> k >> c;
			p[x].emplace_back(k, c);
		}
		init(1);
		gettop(1, 1);
	}
	void init(int v) {
		dep[v] = 1;
		for (auto u : son[v]) {
			init(u);
			if (dep[u] + 1 > dep[v]) {
				dep[v] = dep[u] + 1;
				wson[v] = u;
			}
		}
	}
	void gettop(int v, int t) {
		top[v] = t;
		if (wson[v]) {
			gettop(wson[v], t);
		}
		for (auto u : son[v]) {
			if (u != wson[v]) {
				gettop(u, u);
			}
		}
	}
	ll solve() {
		Segt tr(dep[1]);
		solve(1, tr);
		return max(0ll, tr.getmx(0, n + 1));
	}
	void solve(int v, Segt &tr) {
		if (wson[v]) {
			solve(wson[v], tr);
		}
		int l = dep[top[v]] - dep[v];
		tr.add(l, l, a[v] + max(0ll, tr.getmx(l + 1, n + 1)));
		for (auto u : son[v]) {
			if (u != wson[v]) {
				Segt tx(dep[u]);
				solve(u, tx);
				//cout << u << ": ";
				//tx.print();
				for (int i = 0; i < dep[u]; i++) {
					ll vx = tx.getmx(i, i) + max(0ll, tr.getmx(l + i + 1, n + 1));
					ll vold = tr.getmx(l + i + 1, l + i + 1);
					ll vr = vold + max(0ll, tx.getmx(i, n + 1));
					tr.add(l + i + 1, l + i + 1, max(vx, vr) - vold);
				}
				tr.add(l, l, max(0ll, tx.getmx(0, n + 1)));
			}
		}
		for (auto [k, c] : p[v]) {
			tr.add(l, l + k, -c);
		}
	}
};

void solve() {
	int n, m;
	cin >> n >> m;
	Tree tr(n, m);
	cout << tr.solve() << '\n';
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr), cout.tie(nullptr);
	int T;
	cin >> T;
	while (T--) {
		solve();
	}
	return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3776kb

input:

1
6 3
1 1 2 2 3
2 5 4 3 3 2
2 1 3
3 1 7
1 2 4

output:

6

result:

ok 1 number(s): "6"

Test #2:

score: 0
Accepted
time: 813ms
memory: 94524kb

input:

1003
100 100
1 2 3 4 5 6 7 8 9 10 6 1 2 4 1 11 17 14 17 2 13 8 8 5 11 7 18 6 2 10 23 11 13 3 9 1 33 20 3 9 32 35 11 41 42 29 33 45 21 35 9 36 12 54 19 24 57 31 32 5 3 10 46 15 46 48 20 44 5 41 67 7 18 30 27 6 29 69 57 75 62 74 18 64 17 21 38 60 79 69 54 90 83 83 31 96 31 93 53
152 498 653 559 287 38...

output:

20180
17083
14650
19924
15814
20189
20894
18175
18478
13758
20217
23680
15562
12882
18046
18132
20548
17000
23734
18740
24814
16728
20979
19727
16450
21717
15739
22081
17803
23024
14820
21503
23497
15804
18097
17197
21236
21286
14250
11632
18335
12317
16313
22840
18583
15245
19331
25978
22388
17091
...

result:

ok 1003 numbers