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

提出ソース

結果

問題	No.1030 だんしんぐぱーりない
ユーザー	FF256grhy
提出日時	2020-04-26 02:17:51
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	490 ms / 2,000 ms
コード長	5,815 bytes
コンパイル時間	2,414 ms
コンパイル使用メモリ	180,544 KB
実行使用メモリ	22,016 KB
最終ジャッジ日時	2024-04-26 05:45:19
合計ジャッジ時間	15,944 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	345 ms 19,456 KB
testcase_06	AC	260 ms 15,488 KB
testcase_07	AC	177 ms 8,320 KB
testcase_08	AC	184 ms 10,496 KB
testcase_09	AC	276 ms 19,584 KB
testcase_10	AC	135 ms 5,888 KB
testcase_11	AC	290 ms 11,648 KB
testcase_12	AC	315 ms 17,792 KB
testcase_13	AC	272 ms 16,000 KB
testcase_14	AC	350 ms 15,872 KB
testcase_15	AC	157 ms 5,376 KB
testcase_16	AC	312 ms 13,824 KB
testcase_17	AC	281 ms 20,096 KB
testcase_18	AC	400 ms 16,512 KB
testcase_19	AC	219 ms 7,680 KB
testcase_20	AC	246 ms 12,032 KB
testcase_21	AC	228 ms 14,848 KB
testcase_22	AC	242 ms 12,288 KB
testcase_23	AC	289 ms 11,264 KB
testcase_24	AC	216 ms 6,912 KB
testcase_25	AC	259 ms 11,136 KB
testcase_26	AC	178 ms 5,376 KB
testcase_27	AC	200 ms 5,376 KB
testcase_28	AC	336 ms 14,208 KB
testcase_29	AC	228 ms 17,536 KB
testcase_30	AC	219 ms 12,800 KB
testcase_31	AC	235 ms 11,520 KB
testcase_32	AC	286 ms 15,744 KB
testcase_33	AC	304 ms 18,176 KB
testcase_34	AC	129 ms 6,656 KB
testcase_35	AC	490 ms 21,888 KB
testcase_36	AC	456 ms 21,972 KB
testcase_37	AC	454 ms 21,888 KB
testcase_38	AC	484 ms 22,016 KB
testcase_39	AC	455 ms 21,888 KB
testcase_40	AC	2 ms 5,376 KB
testcase_41	AC	2 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
using LL = long long int;
#define incID(i, l, r) for(int i = (l)    ; i <  (r); ++i)
#define decID(i, l, r) for(int i = (r) - 1; i >= (l); --i)
#define incII(i, l, r) for(int i = (l)    ; i <= (r); ++i)
#define decII(i, l, r) for(int i = (r)    ; i >= (l); --i)
#define inc(i, n)  incID(i, 0, n)
#define dec(i, n)  decID(i, 0, n)
#define inc1(i, n) incII(i, 1, n)
#define dec1(i, n) decII(i, 1, n)
#define inID(v, l, r) ((l) <= (v) && (v) <  (r))
#define inII(v, l, r) ((l) <= (v) && (v) <= (r))
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define FI first
#define SE second
#define FR front()
#define BA back()
#define ALL(v) v.begin(), v.end()
#define RALL(v) v.rbegin(), v.rend()
auto setmin   = [](auto & a, auto b) { return (b <  a ? a = b, true : false); };
auto setmax   = [](auto & a, auto b) { return (b >  a ? a = b, true : false); };
auto setmineq = [](auto & a, auto b) { return (b <= a ? a = b, true : false); };
auto setmaxeq = [](auto & a, auto b) { return (b >= a ? a = b, true : false); };
#define SI(v) static_cast<int>(v.size())
#define RF(e, v) for(auto & e: v)
#define until(e) while(! (e))
#define if_not(e) if(! (e))
#define ef else if
#define UR assert(false)
#define IN(T, ...) T __VA_ARGS__; IN_(__VA_ARGS__);
void IN_() { };
template<typename T, typename ... U> void IN_(T & a, U & ... b) { cin >> a; IN_(b ...); };
template<typename T> void OUT(T && a) { cout << a << endl; }
template<typename T, typename ... U> void OUT(T && a, U && ... b) { cout << a << " "; OUT(b ...); }

// ---- ----

#define bit(b, i) (((b) >> (i)) & 1)
#define PC __builtin_popcountll
#define BL(a) (a ? 64 - __builtin_clzll(a) : 0)

class Forest {
private:
	int n, B;
	vector<vector<int>> g, p;
	vector<int> h, f;
	bool ok;
	void dfs(int v, int pp, int hh) {
		if(f[v] != 0) { return; }
		f[v] = 1;
		p[v][0] = pp;
		h[v] = hh;
		RF(e, g[v]) { dfs(e, v, hh + 1); }
	}
public:
	Forest(int nn = 0) { init(nn); }
	void init(int nn) {
		n = nn;
		B = 0;
		while((1 << B) < n) { B++; }
		g = vector<vector<int>>(n);
		p = vector<vector<int>>(n, vector<int>(B + 1));
		h = vector<int>(n);
		f = vector<int>(n);
		ok = false;
	}
	void add(int a, int b) {
		assert(! ok);
		assert(inID(a, 0, n));
		assert(inID(b, 0, n));
		g[a].PB(b);
		g[b].PB(a);
	}
	void calc(const vector<int> & root = { }) {
		assert(! ok);
		RF(e, root) { dfs(e, e, 0); }
		inc(i, n)   { dfs(i, i, 0); }
		inc(k, B) {
		inc(i, n) {
			p[i][k + 1] = p[p[i][k]][k];
		}
		}
		ok = true;
	}
	bool same(int a, int b) {
		assert(ok);
		assert(inID(a, 0, n));
		assert(inID(b, 0, n));
		return (p[a][B] == p[b][B]);
	}
	int lca(int a, int b) {
		if(! same(a, b)) { return -1; }
		int d = abs(h[a] - h[b]);
		if(h[a] > h[b]) { a = up(a, d); }
		if(h[a] < h[b]) { b = up(b, d); }
		assert(h[a] == h[b]);
		if(a == b) { return a; }
		dec(k, B) {
			if(p[a][k] != p[b][k]) {
				a = p[a][k];
				b = p[b][k];
			}
		}
		assert(a != b && p[a][0] == p[b][0]);
		return p[a][0];
	}
	int dist(int a, int b) {
		if(! same(a, b)) { return -1; }
		return h[a] + h[b] - 2 * h[lca(a, b)];
	}
	int up(int a, int x) {
		assert(ok);
		assert(inID(a, 0, n));
		if(x > h[a]) { return -1; }
		inc(k, B) { if(bit(x, k) == 1) { a = p[a][k]; } }
		return a;
	}
	const vector<vector<int>> & get_g() { return g; }
	const vector<vector<int>> & get_p() { return p; }
	const vector<int>         & get_h() { return h; }
};

// ----

template<typename T> class SegmentTree {
private:
	int n, s;
	vector<T> a;
	function<T(T &, T &)> f;
	T e;
	bool ok;
public:
	SegmentTree() { n = 0; }
	SegmentTree(int nn, function<T(T &, T &)> ff, T ee) { init(nn, ff, ee); }
	void init(int nn, function<T(T &, T &)> ff, T ee) {
		n = nn;
		f = ff;
		e = ee;
		s = 1;
		while(s < n) { s *= 2; }
		a = vector<T>(2 * s, e);
		ok = true;
	}
	void shift(int & p) {
		assert(inID(p, 0, n));
		p += s;
	}
	void apply(int p, function<void(T &)> g) {
		shift(p);
		g(a[p]);
		while(p > 1) {
			p /= 2;
			a[p] = f(a[2 * p], a[2 * p + 1]);
		}
	}
	T fold_ID(int l, int r) {
		assert(ok);
		assert(inII(l, 0, n)); l += s;
		assert(inII(r, 0, n)); r += s; r--;
		T x = e, y = e;
		while(l <= r) {
			if(l % 2 == 1) { x = f(x, a[l]); l++; }
			if(r % 2 == 0) { y = f(a[r], y); r--; }
			l /= 2;
			r /= 2;
		}
		return f(x, y);
	}
	T fold_II(int l, int r) { return fold_ID(l + 0, r + 1); }
	T fold_CI(int l, int r) { return fold_ID(l + 1, r + 1); }
	T fold_CD(int l, int r) { return fold_ID(l + 1, r + 0); }
	const T & operator[](int p) {
		shift(p);
		return a[p];
	}
	T & ref(int p) {
		shift(p);
		ok = false;
		return a[p];
	}
	void update() {
		dec(i, s) { a[i] = f(a[2 * i], a[2 * i + 1]); }
		ok = true;
	}
};
#define OP(s) [&](auto & A, auto & B) { return s; }
#define AP(s) [&](auto & A) { s; }

// ----

template<typename T> istream & operator>>(istream & s, vector<T> & v) { RF(e, v) { s >> e; } return s; }
template<typename T> ostream & operator<<(ostream & s, vector<T> const & v) {
	inc(i, SI(v)) { s << (i == 0 ? "" : " ") << v[i]; }
	return s;
}

int main() {
	IN(int, n, k, Q);
	vector<LL> c(n);
	cin >> c;
	vector<int> a(k);
	RF(e, a) { cin >> e; e--; }
	Forest tree(n);
	inc(i, n - 1) {
		IN(int, a, b);
		a--; b--;
		tree.add(a, b);
	}
	tree.calc();
	
	auto f = [&](int x, int y) {
		if(x == -1 && y == -1) { return -1; }
		if(x == -1) { return y; }
		if(y == -1) { return x; }
		return tree.lca(x, y);
	};
	SegmentTree<int> st(k, f, -1);
	inc(i, k) { st.ref(i) = a[i]; }
	st.update();
	
	function<void(int, int)> dfs = [&](int v, int p) {
		if(p != -1) { setmax(c[v], c[p]); }
		RF(e, tree.get_g()[v]) {
			if(e == p) { continue; }
			dfs(e, v);
		}
	};
	dfs(0, -1);
	
	inc(q, Q) {
		IN(int, t, x, y);
		x--; y--;
		if(t == 1) {
			a[x] = y;
			st.apply(x, AP(A = y));
		} else {
			OUT(c[st.fold_II(x, y)]);
		}
	}
}

yukicoder

結果

テストケース

ソースコード