結果

問題 No.898 tri-βutree
ユーザー kuhaku
提出日時 2019-10-11 18:56:44
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,482 bytes
コンパイル時間 2,115 ms
コンパイル使用メモリ 176,476 KB
実行使用メモリ 38,172 KB
最終ジャッジ日時 2024-11-25 01:32:40
合計ジャッジ時間 22,346 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other WA * 20 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define REP(i, n) for(int (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<int, int> P;

#define MAX_N 100000

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

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

int lca(int a, int b){
	int 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){
	int 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));
		}
	}

	int q;
	cin >> q;

	REP(i, q){
		int 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;
}
0