#388589 (C++14) No.898 tri-βutree

• 問題: No.898 tri-βutree
• ユーザー: kuhaku
• 提出日時: 2019-10-11 18:58:50
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: TLE
• 実行時間: -
• コード長: 1,469 bytes
• コンパイル時間: 1,587 ms
• コンパイル使用メモリ: 176,244 KB
• 実行使用メモリ: 38,824 KB
• 最終ジャッジ日時: 2024-11-25 01:41:18
• 合計ジャッジ時間: 18,240 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	TLE	-
testcase_01	AC	3 ms; 6,816 KB
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
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	-

ソースコード

#include <bits/stdc++.h>
using namespace std;

#define REP(i, n) for(ll (i) = 0; (i) < (n); (i)++)
#define MAX(a, b) ((a) > (b) ? (a) : (b))
#define MIN(a, b) ((a) < (b) ? (a) : (b))
#define mp make_pair
#define INF 2e9

typedef long long ll;
typedef pair<ll, ll> P;

#define MAX_N 100000

vector<P> tree[MAX_N];
ll dist[MAX_N], parent[MAX_N], depth[MAX_N];

void dfs(ll n, ll p){
	REP(i, tree[n].size()){
		ll node = tree[n][i].first;
		if(node != p){
			parent[node] = n;
			depth[node] = depth[n]+1;
			dfs(node, n);
		}
	}
}

ll lca(ll a, ll b){
	ll m = a, n = b;
	while(m != n){
		if(depth[m] >= depth[n]) m = parent[m];
		if(depth[n] > depth[m]) n = parent[n];
	}

	return m;
}

int main(void){
	ll n;
	cin >> n;

	fill(dist, dist+n, INF);

	REP(i, n-1){
		ll u, v, w;
		cin >> u >> v >> w;
		tree[u].push_back(mp(v, w));
		tree[v].push_back(mp(u, w));
	}

	dfs(0, 0);

	priority_queue<P, vector<P>, greater<P> > que;
	dist[0] = 0;
	que.push(mp(0, 0));
	while(!que.empty()){
		P out = que.top(); que.pop();
		if(out.first > dist[out.second]) continue;
		dist[out.second] = out.first;
		REP(i, tree[out.second].size()){
			P in = tree[out.second][i];
			que.push(mp(out.first+in.second, in.first));
		}
	}

	ll q;
	cin >> q;

	REP(i, q){
		ll x, y, z;
		cin >> x >> y >> z;

		ll ans = 0;
		ans += dist[x]+dist[y]-2*dist[lca(x, y)];
		ans += dist[y]+dist[z]-2*dist[lca(y, z)];
		ans += dist[z]+dist[x]-2*dist[lca(z, x)];

		cout << ans/2 << endl;
	}

	return 0;
}