#456602 (C++14) No.1030 だんしんぐぱーりない

• 問題: No.1030 だんしんぐぱーりない
• ユーザー: Rho
• 提出日時: 2020-04-02 14:16:32
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 2,261 bytes
• コンパイル時間: 2,028 ms
• コンパイル使用メモリ: 181,304 KB
• 実行使用メモリ: 19,652 KB
• 最終ジャッジ日時: 2024-07-03 00:41:54
• 合計ジャッジ時間: 20,628 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	5 ms; 8,192 KB
testcase_01	AC	5 ms; 8,192 KB
testcase_02	AC	5 ms; 8,192 KB
testcase_03	WA	-
testcase_04	AC	5 ms; 8,192 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-
testcase_36	WA	-
testcase_37	WA	-
testcase_38	WA	-
testcase_39	WA	-
testcase_40	AC	5 ms; 8,192 KB
testcase_41	AC	4 ms; 8,192 KB

ソースコード

#include "bits/stdc++.h"
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define int long long
vector<int>G[200006];
int d[200006];
int par2[20][200006];
int a[100006],c[100006];

void dfs(int now, int par) {
	par2[0][now] = par;
	for (int e : G[now]) {
		if (e == par)continue;
		d[e] = d[now] + 1;
		c[e] = max(c[now], c[e]);
		dfs(e, now);
	}
}

int lca(int u, int v) {
	if (d[u] > d[v])swap(u, v);
	rep(k, 20) {
		if ((d[v] - d[u]) >> k & 1) {
			v = par2[k][v];
		}
	}
	if (u == v)return u;
	for (int k = 19; k >= 0; k--) {
		if (par2[k][u] != par2[k][v]) {
			u = par2[k][u]; v = par2[k][v];
		}
	}
	return par2[0][u];
}
struct segtree {
	vector<int>node;
	int n;

	void init(int N) {
		n = 1;
		while (n < N)n *= 2;
		node.resize(2 * n, -1);
	}

	void upd(int i, int x) {
		i += n - 1;
		node[i] = x;
		while (i) {
			i = (i - 1) / 2;
			int x1 = node[2 * i + 1], x2 = node[2 * i + 2];
			if (x1 == -1 && x2 == -1)node[i] = -1;
			else if (x1 == -1)node[i] = x2;
			else if (x2 == -1)node[i] = x1;
			else node[i] = lca(x1, x2);
		}

	}

	int calc(int a, int b, int k, int l, int r) {
		if (r <= a || b <= l)return -1;
		if (a <= l&&r <= b)return node[k];
		int vl = calc(a, b, k * 2 + 1, l, (l + r) / 2);
		int vr = calc(a, b, k * 2 + 2, (l + r) / 2, r);
		if (vl == -1 && vr == -1)return -1;
		else if (vl == -1)return vr;
		else if(vr == -1)return vl;
		else return lca(vl, vr);
	}

	int get(int a, int b) {
		return calc(a, b, 0, 0, n);
	}

};
segtree seg;
signed main() {
	int n, k, q; cin >> n >> k >> q;
	assert(1 <= n&&n <= 100000 && 1 <= k&&k <= 100000 && 1 <= q&&q <= 100000);
	rep(i, n) {
		cin >> c[i];
		assert(1 <= c[i] && c[i] <= 1000000000);
	}
	rep(i, k) {
		cin >> a[i];
		assert(1 <= a[i] && a[i] <= n);
	}
	rep(i, n - 1) {
		int a, b; cin >> a >> b;
		assert(1 <= a&&a <= n && 1 <= b&&b <= n);
		a--; b--;
		G[a].push_back(b); G[b].push_back(a);
	}
	
	dfs(0, -1);
	seg.init(k);
	rep(i, k)seg.upd(i, a[i]-1);

	rep(i, q) {
		int t; cin >> t;
		if (t == 1) {//変更
			int b, x; cin >> b >> x;
			assert(1 <= b&&b <= k && 1 <= x&&x <= n);
			seg.upd(b - 1, x - 1);
		}
		else if(t==2){//集会
			int l, r; cin >> l >> r;
			assert(1 <= l&&l <= r&&r <= k);
			cout << c[seg.get(l - 1, r)] << endl;
		}
		else assert(0);
	}
}