結果

問題 No.901 K-ary εxtrεεmε
ユーザー 👑 NachiaNachia
提出日時 2020-09-27 19:55:49
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 333 ms / 3,000 ms
コード長 1,392 bytes
コンパイル時間 1,773 ms
コンパイル使用メモリ 178,372 KB
実行使用メモリ 24,356 KB
最終ジャッジ日時 2024-06-30 12:53:59
合計ジャッジ時間 9,237 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 146 ms
24,356 KB
testcase_01 AC 5 ms
13,852 KB
testcase_02 AC 7 ms
13,728 KB
testcase_03 AC 6 ms
13,964 KB
testcase_04 AC 7 ms
14,048 KB
testcase_05 AC 6 ms
14,472 KB
testcase_06 AC 7 ms
15,320 KB
testcase_07 AC 214 ms
20,768 KB
testcase_08 AC 215 ms
20,640 KB
testcase_09 AC 220 ms
20,768 KB
testcase_10 AC 215 ms
20,764 KB
testcase_11 AC 216 ms
20,768 KB
testcase_12 AC 200 ms
20,764 KB
testcase_13 AC 208 ms
20,772 KB
testcase_14 AC 209 ms
20,768 KB
testcase_15 AC 206 ms
20,896 KB
testcase_16 AC 199 ms
20,768 KB
testcase_17 AC 292 ms
20,772 KB
testcase_18 AC 289 ms
20,764 KB
testcase_19 AC 284 ms
20,760 KB
testcase_20 AC 286 ms
20,764 KB
testcase_21 AC 281 ms
20,640 KB
testcase_22 AC 204 ms
20,900 KB
testcase_23 AC 208 ms
20,892 KB
testcase_24 AC 209 ms
20,900 KB
testcase_25 AC 207 ms
20,888 KB
testcase_26 AC 202 ms
20,888 KB
testcase_27 AC 333 ms
20,768 KB
testcase_28 AC 333 ms
20,900 KB
testcase_29 AC 329 ms
20,776 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using LL=long long;
using ULL=unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

const int xN=100000;

struct Edge{ int to; LL dist; };

int N;
vector<Edge> E[xN];
int P[xN];
LL D[xN];
int dfsI[xN];
int lcaP[20][xN];
int lcaD[xN];

int dfs(int p=0,int i=0){
	dfsI[p] = i++;
	for(Edge& e:E[p]){
		if(P[p]==e.to) continue;
		D[e.to] = D[p] + e.dist;
		lcaD[e.to] = lcaD[p] + 1;
		P[e.to] = p;
		i = dfs(e.to,i);
	}
	return i;
}

void init(){
	rep(i,N) P[i]=-1;
	rep(i,N) D[i]=0;
	dfs(0);

	rep(i,N) lcaP[0][i] = P[i];
	lcaP[0][0] = 0;
	rep(d,19) rep(i, N) lcaP[d + 1][i] = lcaP[d][lcaP[d][i]];
}

int LCA(int u,int v){
	if(lcaD[u]>lcaD[v]) swap(u,v);
	for(int i=19; i>=0; i--) if(lcaD[u] + (1<<i) <= lcaD[v]) v=lcaP[i][v];
	if(u==v) return u;
	for(int i=19; i>=0; i--) if(lcaP[i][u] != lcaP[i][v]){
		u=lcaP[i][u];
		v=lcaP[i][v];
	}
	return lcaP[0][u];
}

LL distance(int u,int v){
	int g=LCA(u,v);
	return D[u]+D[v]-2*D[g];
}

int main(){
	cin>>N;
	rep(i,N-1){
		int u,v,c; cin>>u>>v>>c;
		E[u].push_back({v,c});
		E[v].push_back({u,c});
	}
	init();

	int Q; cin>>Q;
	while(Q--){
		int K; cin>>K;
		vector<pair<int,int>> X(K);
		rep(i,K){ int x; cin>>x; X[i] = {dfsI[x],x}; }
		sort(X.begin(),X.end());
		X.push_back(X.front());
		LL ans=0;
		rep(i,K) ans += distance(X[i].second,X[i+1].second);
		cout<<(ans/2)<<endl;
	}

	return 0;
}
0