結果

問題 No.650 行列木クエリ
ユーザー DaylightDaylight
提出日時 2023-03-12 15:24:17
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 573 ms / 2,000 ms
コード長 5,250 bytes
コンパイル時間 5,270 ms
コンパイル使用メモリ 286,688 KB
実行使用メモリ 135,916 KB
最終ジャッジ日時 2023-10-18 10:37:26
合計ジャッジ時間 8,599 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 218 ms
31,780 KB
testcase_02 AC 573 ms
125,884 KB
testcase_03 AC 2 ms
4,348 KB
testcase_04 AC 225 ms
31,780 KB
testcase_05 AC 564 ms
125,884 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 215 ms
33,628 KB
testcase_09 AC 504 ms
135,916 KB
testcase_10 AC 2 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>

#include <bits/stdc++.h>

using namespace std;
#define SZ(x) (int) (x).size()
#define REP(i, n) for(int i = 0; i < (n); i++)
#define FOR(i, a, b) for(auto i = (a); i < (b); i++)
#define For(i, a, b, c) \
	for(auto i = (a); i != (b); i += (c))
#define REPR(i, n) for(auto i = (n) -1; i >= 0; i--)
#define ALL(s) (s).begin(), (s).end()
#define so(V) sort(ALL(V))
#define rev(V) reverse(ALL(V))
#define uni(v) v.erase(unique(ALL(v)), (v).end())
#define eb emplace_back

typedef unsigned long long ull;
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<bool> vb;
typedef vector<vi> vvi;
typedef vector<vll> vvll;
typedef pair<int, int> PI;
typedef pair<ll, ll> PL;
const double EPS = 1e-6;
const int MOD = 1e9 + 7;
const int INF = (1 << 30);
const ll LINF = 1e18;
const double math_PI = acos(-1);

template<typename T>
vector<T> make_v(size_t a) {
	return vector<T>(a);
}

template<typename T, typename... Ts>
auto make_v(size_t a, Ts... ts) {
	return vector<decltype(make_v<T>(ts...))>(
		a, make_v<T>(ts...));
}

template<typename T, typename V>
typename enable_if<is_class<T>::value == 0>::type fill_v(
	T& t, const V& v) {
	t = v;
}

template<typename T, typename V>
typename enable_if<is_class<T>::value != 0>::type fill_v(
	T& t, const V& v) {
	for(auto& e: t) fill_v(e, v);
}

template<class T>
bool chmax(T& a, const T& b) {
	if(a < b) {
		a = b;
		return true;
	}
	return false;
}

template<class T>
bool chmin(T& a, const T& b) {
	if(a > b) {
		a = b;
		return true;
	}
	return false;
}

template<typename S, typename T>
istream& operator>>(istream& is, pair<S, T>& p) {
	cin >> p.first >> p.second;
	return is;
}

template<typename T>
istream& operator>>(istream& is, vector<T>& vec) {
	for(T& x: vec) is >> x;
	return is;
}

template<typename T>
string join(vector<T>& vec, string splitter) {
	stringstream ss;
	REP(i, SZ(vec)) {
		if(i != 0) ss << splitter;
		ss << vec[i];
	}
	return ss.str();
}

template<typename T>
ostream& operator<<(ostream& os, vector<T>& vec) {
	os << join(vec, " ");
	return os;
}
class HLdecomposition {
private:
	int V;
	vector<vector<int>> G;
	vector<int> stsize, parent, pathtop, in, out;
	int root;
	void BuildStsize(int u, int p) {
		stsize[u] = 1, parent[u] = p;
		for(int& v: G[u]) {
			if(v == p) {
				if(v == G[u].back())
					break;
				else
					swap(v, G[u].back());
			}
			BuildStsize(v, u);
			stsize[u] += stsize[v];
			if(stsize[v] > stsize[G[u][0]]) {
				swap(v, G[u][0]);
			}
		}
	}
	void BuildPath(int u, int p, int& tm) {
		in[u] = tm++;
		for(int v: G[u]) {
			if(v == p) continue;
			pathtop[v] = (v == G[u][0] ? pathtop[u] : v);
			BuildPath(v, u, tm);
		}
		out[u] = tm;
	}

public:
	void add_edge(int u, int v) {
		G[u].push_back(v), G[v].push_back(u);
	}
	void build(int _root = 0) {
		root = _root;
		int tm = 0;
		BuildStsize(root, -1);
		pathtop[root] = root;
		BuildPath(root, -1, tm);
	}
	//蜈・・鬆らせ縺ｮ繧､繝ｳ繝・ャ繧ｯ繧ｹ縺ｮ驟榊・荳翫〒縺ｮid繧定ｿ斐☆
	inline int get(int a) {
		return in[a];
	}
	int lca(int a, int b) {
		int pa = pathtop[a], pb = pathtop[b];
		while(pathtop[a] != pathtop[b]) {
			if(in[pa] > in[pb]) {
				a = parent[pa], pa = pathtop[a];
			} else {
				b = parent[pb], pb = pathtop[b];
			}
		}
		if(in[a] > in[b]) swap(a, b);
		return a;
	}
	void subtree_query(
		int a, const function<void(int, int)>& func) {
		func(in[a], out[a]);
	}
	// 萓・ hl.path_query(q, r, [&](int l, int r){ seg.range(l, r, s); })
	void path_query(int a, int b,
					const function<void(int, int)>& func) {
		int pa = pathtop[a], pb = pathtop[b];
		while(pathtop[a] != pathtop[b]) {
			if(in[pa] > in[pb]) {
				func(in[pa], in[a] + 1);
				a = parent[pa], pa = pathtop[a];
			} else {
				func(in[pb], in[b] + 1);
				b = parent[pb], pb = pathtop[b];
			}
		}
		if(in[a] > in[b]) swap(a, b);
		func(in[a], in[b] + 1);
	}
	HLdecomposition(int node_size)
		: V(node_size),
		  G(V),
		  stsize(V, 0),
		  parent(V, -1),
		  pathtop(V, -1),
		  in(V, -1),
		  out(V, -1) {
	}
};
using namespace atcoder;
using mint = modint1000000007;
using S = vector<vector<mint>>;
S op(S a, S b) {
	S c = S(2, vector<mint>(2));
	REP(i, 2) {
		REP(j, 2) {
			REP(k, 2) {
				c[i][j] += a[i][k] * b[k][j];
			}
		}
	}
	return c;
}
S e() {
	S c = S(2, vector<mint>(2));
	c[0][0] = 1;
	c[1][1] = 1;
	return c;
}
int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int N;
	cin >> N;
	HLdecomposition hld(N + N - 1);
	REP(i, N - 1) {
		int A, B;
		cin >> A >> B;
		hld.add_edge(A, N + i);
		hld.add_edge(N + i, B);
	}
	hld.build();
	segtree<S, op, e> seg(2 * N - 1);
	int Q;
	cin >> Q;
	REP(q, Q) {
		char c;
		cin >> c;
		if(c == 'x') {
			int i;
			cin >> i;
			int x1, x2, x3, x4;
			cin >> x1 >> x2 >> x3 >> x4;
			seg.set(hld.get(N + i),
					vector<vector<mint>> {
						{ mint(x1), mint(x2) },
						{ mint(x3), mint(x4) } });
		} else {
			int i, j;
			cin >> i >> j;
			auto c = e();
			vector<PI> V;
			hld.path_query(i, j, [&](int l, int r) {
				V.eb(l, r);
			});
			rev(V);
			for(auto [l, r]: V) {
				c = op(c, seg.prod(l, r));
			}
			cout << c[0][0].val() << " " << c[0][1].val()
				 << " " << c[1][0].val() << " "
				 << c[1][1].val() << endl;
		}
	}
	return 0;
}
0