QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#204056#7563. Fun on TreeForever_Young#RE 0ms0kbC++145.7kb2023-10-07 00:36:592023-10-07 00:37:00

Judging History

你现在查看的是最新测评结果

  • [2023-10-07 00:37:00]
  • 评测
  • 测评结果:RE
  • 用时:0ms
  • 内存:0kb
  • [2023-10-07 00:36:59]
  • 提交

answer

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define all(x) ((x).begin()), ((x).end())
#define fi first
#define se second
typedef long long LL;
const LL infll = 1e18;
const int N = 200022;
int vst[N], fa[N], siz[N], p[N];
int tim = 0;
LL w[N], a[N];
bool banned[N];
vector<pair<int, LL> > e[N];
struct SGT {
	int n;
	vector<pair<LL, int>> a;
	void build(int k, int le, int ri) {
		a[k].se = -1;
		a[k].fi = -infll;
		if(le != ri) {
			int mid = (le + ri) / 2;
			build(k * 2, le, mid);
			build(k * 2 + 1, mid + 1, ri);
		}
	}
	void modify(int k, int le, int ri, int pos, pair<LL, int> val) {
		if(le == ri) {
			a[k] = val;
		}else {
			int mid = (le + ri) / 2;
			if(pos <= mid) {
				modify(k * 2, le, mid, pos, val);
			}else {
				modify(k * 2 + 1, mid + 1, ri, pos, val);
			}
			a[k] = max(a[k * 2], a[k * 2 + 1]);
		}
	}
	void add(int k, int le, int ri, int pos, LL val) {
		if(le == ri) {
			a[k].fi += val;
		}else {
			int mid = (le + ri) / 2;
			if(pos <= mid) {
				add(k * 2, le, mid, pos, val);
			}else {
				add(k * 2 + 1, mid + 1, ri, pos, val);
			}
			a[k] = max(a[k * 2], a[k * 2 + 1]);
		}
	}
	pair<LL, int> getmax(int k, int le, int ri, int ql, int qr) {
		if(ql > qr) {
			return make_pair(-infll, -1);
		}
		if(ql <= le && ri <= qr) {
			return a[k];
		}else {
			pair<LL, int> res{-infll, -1};
			int mid = (le + ri) / 2;
			if(ql <= mid) res = max(res, getmax(k * 2, le, mid, ql, qr));
			if(qr >= mid + 1) res = max(res, getmax(k * 2 + 1, mid + 1, ri, ql, qr));
			return res;
		}
	}
};
struct Dnode {
	int rt;
	map<int, int> mp;
	vector<int> fa, blg;
	vector<LL> dep;
	LL tag;
	SGT sgt;
	LL len;
	vector<Dnode*> s;
	void resize(int n) {

		fa.resize(n);
		blg.resize(n);
		dep.resize(n);
	}
} dn_pool[N * 2];
int dn_num = 0;
Dnode * dvcq(int st) {
	Dnode &dn = dn_pool[dn_num++];
	map<int, int> &mp(dn.mp);

	vector<int> q;
	q.pb(st);
	mp[st] = q.size();
	tim++;
	vst[st] = tim;
	fa[st] = -1;
	for(int op = 0; op < (int)q.size(); op++) {
		int v = q[op];
		siz[v] = 0;
		if(banned[v]) continue;
		for(auto t : e[v]) {
			int y = t.fi;
			if(vst[y] != tim) {
				vst[y] = tim;
				q.pb(y);
				mp[y] = q.size();
				fa[y] = v;
			}
		}
	}
	for(int i = (int)q.size() - 1; i >= 0; i--) {
		siz[q[i]]++;
		if(fa[q[i]] != -1) siz[fa[q[i]]] += siz[q[i]];
	}
	cout << "q" << q.size() << endl;
	dn.resize(q.size());

	int rt = st;
	
	for(;;) {
		bool flag = true;
		for(auto t : e[rt]) {
			int y = t.fi;
			if(fa[y] == rt && siz[y] > q.size() / 2) {
				rt = y;
				flag = false;
				break;
			}
		}
		if(flag) break;
	}

	if(q.size() == 2) rt = banned[st] ? q.back() : q.front();
	
	q.clear();
	q.pb(rt);
	dn.rt = rt;
	tim++;
	vst[rt] = tim;
	vector<int> & blg(dn.blg);
	vector<int> & fa(dn.fa);
	vector<LL> & dep(dn.dep);
	dep[mp[rt]] = 0;
	fa[mp[rt]] = -1;
	blg[mp[rt]] = -1;
	for(int op = 0; op < (int)q.size(); op++) {
		int v = q[op];
		int _ = 0;
		int mpv = mp[v];
		if(banned[v]) continue;
		for(auto t : e[v]) {
			int y = t.fi;
			if(vst[y] != tim) {
				vst[y] = tim;
				q.pb(y);
				int mpy = mp[y];
				fa[mpy] = mpv;
				dep[mpy] = dep[mpv] + t.se;
				if(v == rt) {
					blg[mpy] = _;
				}else {
					blg[mpy] = blg[mpv];
				}
			}
			_++;
		}
	}
	assert(banned[rt] == false);
	if(q.size() == 2) {
		dn.len = dep[mp[q.back()]];
		dn.sgt.n = 2;
		dn.sgt.a.resize(9);
		dn.sgt.build(1, 1, 2);
		dn.sgt.modify(1, 1, 2, 1, make_pair(a[rt], -rt));
		dn.sgt.modify(1, 1, 2, 2, make_pair(a[q.back()], -q.back()));
		return &dn;
	}
	banned[rt] = true;
	dn.sgt.n = e[rt].size();
	dn.sgt.a.resize(4 * dn.sgt.n + 1);
	dn.sgt.build(1, 1, dn.sgt.n);
	dn.s.resize(dn.sgt.n);
	for(int i = 0; i < (int)e[rt].size(); i++) {
		dn.s[i] = dvcq(e[rt][i].fi);
		dn.sgt.modify(1, 1, dn.sgt.n, i + 1, dn.s[i]->sgt.a[1]);
	}
	dn.tag = 0;
	return &dn;
}
pair<LL, int> operator + (const pair<LL, int> & a, LL b) {
	return make_pair(a.fi + b, a.se);
}
void add(Dnode * p, int y, int fa, LL w) {
	int mpy = p->mp[y];
	int mpp = p->mp.count(fa) ? p->mp[fa] : -1;
	if(p->sgt.a.empty()) {
		if(mpp != -1) {
			int pos = mpy == p->rt ? 1 : 2;
			p->sgt.add(1, 1, 2, pos, w);
		}else {
			p->sgt.add(1, 1, 2, 1, w);
			p->sgt.add(1, 1, 2, 2, w);
		}
		return;
	}
	if(mpp != -1 && p->fa[mpp] == mpy) {
		p->tag += w;
		add(p->s[p->blg[mpp]], fa, y, -w);
		p->sgt.modify(1, 1, p->sgt.n, p->blg[mpp] + 1, p->s[p->blg[mpp]]->sgt.a[1] + p->s[p->blg[mpp]]->tag);
	}else {
		if(y == p->rt) {
			p->tag += w;
		}else {
			add(p->s[p->blg[mpy]], y, fa, w);
			p->sgt.modify(1, 1, p->sgt.n, p->blg[mpy] + 1, p->s[p->blg[mpy]]->sgt.a[1] + p->s[p->blg[mpy]]->tag);
		}
	}
}
pair<LL, int> query(Dnode * p, int x) {
	if(p->sgt.a.empty()) {
		if(x == p->rt) {
			return max(p->sgt.a[2], p->sgt.a[3] + p->len);
		}else {
			return max(p->sgt.a[2] + p->len, p->sgt.a[3]);
		}
	}
	LL mpx = p->mp[x];
	LL depx = p->dep[mpx];
	if(x == p->rt) {
		return p->sgt.a[1] + p->tag;
	}
	pair<LL, int> res{-infll, -1};
	LL blg = p->blg[mpx];
	res = max(p->sgt.getmax(1, 1, p->sgt.n, 1, blg), p->sgt.getmax(1, 1, p->sgt.n, blg + 2, p->sgt.n));
	res = max(res, query(p->s[blg], x));
	return res + p->tag;
}
int main() {
	int n, q;
	scanf("%d%d", &n, &q);
	for(int i = 1; i <= n; i++) {
		scanf("%lld", &a[i]);
	}
	for(int i = 1; i < n; i++) {
		scanf("%d%lld", &p[i + 1], &w[i + 1]);
		e[p[i + 1]].pb({i + 1, w[i + 1]});
		e[i + 1].pb({p[i + 1], w[i + 1]});
	}
	Dnode * rt = dvcq(1);
	for(int i = 1; i <= q; i++) {
		LL x, y, w;
		scanf("%lld%lld%lld", &x, &y, &w);
		add(rt, y, p[y], w);
		pair<LL, int> res = query(rt, x);
		printf("%d %lld\n", res.se * -1, res.fi);
	}
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Runtime Error

input:

6 6
1 1 4 5 1 4
1 5
2 0
3 2
4 1
5 6
3 2 -100000
1 2 100000
1 1 0
2 2 66
3 1 5
4 4 -3

output:

q6
q3

result: