結果

問題 No.901 K-ary εxtrεεmε
ユーザー Jiro_tech15Jiro_tech15
提出日時 2019-10-04 22:13:47
言語 C++11
(gcc 11.4.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 4,167 bytes
コンパイル時間 1,160 ms
コンパイル使用メモリ 102,808 KB
実行使用メモリ 48,120 KB
最終ジャッジ日時 2024-04-15 00:09:32
合計ジャッジ時間 12,407 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 8 ms
21,396 KB
testcase_02 AC 11 ms
20,288 KB
testcase_03 AC 11 ms
20,944 KB
testcase_04 AC 12 ms
20,836 KB
testcase_05 AC 11 ms
21,360 KB
testcase_06 AC 12 ms
20,164 KB
testcase_07 AC 385 ms
41,304 KB
testcase_08 AC 392 ms
41,512 KB
testcase_09 AC 387 ms
41,672 KB
testcase_10 AC 388 ms
41,504 KB
testcase_11 AC 391 ms
42,680 KB
testcase_12 AC 368 ms
42,564 KB
testcase_13 AC 361 ms
41,272 KB
testcase_14 AC 376 ms
41,432 KB
testcase_15 AC 370 ms
42,636 KB
testcase_16 AC 366 ms
41,092 KB
testcase_17 AC 430 ms
41,316 KB
testcase_18 AC 424 ms
41,252 KB
testcase_19 AC 422 ms
42,800 KB
testcase_20 AC 422 ms
41,596 KB
testcase_21 AC 428 ms
42,572 KB
testcase_22 AC 358 ms
41,872 KB
testcase_23 AC 356 ms
41,588 KB
testcase_24 AC 359 ms
41,376 KB
testcase_25 AC 363 ms
41,644 KB
testcase_26 AC 363 ms
42,224 KB
testcase_27 AC 512 ms
42,352 KB
testcase_28 AC 515 ms
41,684 KB
testcase_29 AC 505 ms
41,576 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <string>
#include <math.h>
#include <iomanip>
#include <limits>
#include <list>
#include <queue>
#include <tuple>
#include <map>
#include <stack>
#include <set>
using namespace std;
#define MOD (long long int)(1e9+7)
#define ll long long int
#define rep(i,n) for(int i=0; i<(int)(n); i++)
#define reps(i,n) for(int i=1; i<=(int)(n); i++)
#define REP(i,n) for(int i=n-1; i>=0; i--)
#define REPS(i,n) for(int i=n; i>0; i--)
#define INF (int)(1123456789)
#define LINF (long long int)(112345678901234567)
#define chmax(a, b) a = (((a)<(b)) ? (b) : (a))
#define chmin(a, b) a = (((a)>(b)) ? (b) : (a))
#define all(v) v.begin(), v.end()

const int N = (int)3e5;

ll mpow(ll a, ll b){
  if(b==0) return 1;
  else if(b%2==0){ll memo = mpow(a,b/2); return memo*memo%MOD;}
  else return mpow(a,b-1) * a % MOD;
}
ll lpow(ll a, ll b){
  if(b==0) return 1;
  else if(b%2==0){ll memo = lpow(a,b/2); return memo*memo;}
  else return lpow(a,b-1) * a;
}
ll gcd(ll a, ll b){
  if(b==0) return a;
  else return gcd(b, a%b);
}
vector<ll> kaijo_memo;
ll kaijo(ll n){
  if(kaijo_memo.size() > n) return kaijo_memo[n];
  if(kaijo_memo.size() == 0) kaijo_memo.push_back(1);
  while(kaijo_memo.size() <= n) kaijo_memo.push_back(kaijo_memo[kaijo_memo.size()-1] * kaijo_memo.size() % MOD);
  return kaijo_memo[n];
}
vector<ll> gyaku_kaijo_memo;
ll gyaku_kaijo(ll n){
  if(gyaku_kaijo_memo.size() > n) return gyaku_kaijo_memo[n];
  if(gyaku_kaijo_memo.size() == 0) gyaku_kaijo_memo.push_back(1);
  while(gyaku_kaijo_memo.size() <= n) gyaku_kaijo_memo.push_back(gyaku_kaijo_memo[gyaku_kaijo_memo.size()-1] * mpow(gyaku_kaijo_memo.size(), MOD-2) % MOD);
  return gyaku_kaijo_memo[n];
}

ll nCr(ll n, ll r){
  if(n == r) return 1;//0個の丸と-1個の棒みたいな時に時に効く？不安.
  if(n < r || r < 0) return 0;
  ll ret = 1;
  ret *= kaijo(n); ret %= MOD;
  ret *= gyaku_kaijo(r); ret %= MOD;
  ret *= gyaku_kaijo(n-r); ret %= MOD;
  return ret;
}

struct edge{ll cost; int to;};
vector<edge> G[N];
vector<int> P[N];
bool checked[N];
int number[N],first[N],last[N],D[N];
vector<int> tour;

int dfs(int now, int num, int depth){
  D[now] = depth;
  checked[now] = true;
  number[now] = num;
  num++;
  first[now] = tour.size();
  rep(i,G[now].size()){
    int next = G[now][i].to;
    if(checked[next]) continue;
    tour.push_back(G[now][i].cost);
    P[next].push_back(now);
    int ret = dfs(next, num, depth+1);
    if(ret == -1) continue;
    num = ret;
    tour.push_back(G[now][i].cost * -1);
  }
  return num;
}

int lca(int a, int b){
  REP(t,30){
    if(D[a] - lpow(2,t) >= D[b]){
      a = P[a][t];
    }else if(D[b] - lpow(2,t) >= D[a]){
      b = P[b][t];
    }
  }

  REP(t,30){
    if(P[a][t] != P[b][t]){
      a = P[a][t];
      b = P[b][t];
    }
  }
  if(a != b){
    a = P[a][0];
    b = P[b][0];
  }
  if(a != b){
    cout<<"おかしいよ！"<<endl;
    while(true){

    }
  }
  return a;
}

int main(void){
  int n;cin>>n;
  rep(i,n-1){
    int u,v,w;
    cin>>u>>v>>w;
    G[u].push_back({w,v});
    G[v].push_back({w,u});
  }

  rep(i,n){
    checked[i] = false;
  }

  P[0].push_back(-1);
  tour.push_back(0);
  dfs(0, 0, 0);
  rep(t,30){
    rep(i,n){
      if(P[i][t] == -1){
        P[i].push_back(-1);
        continue;
      }
      P[i].push_back(P[P[i][t]][t]);
    }
  }

  /*rep(i,n){
    cout<<number[i]<<" ";
  }
  cout<<endl;

  rep(i,tour.size()){
    cout<<tour[i]<<" ";
  }
  cout<<endl;

  rep(i,n){
    cout<<P[i][1]<<" ";
  }
  cout<<endl;*/

  rep(i,tour.size()-1){
    tour[i+1] += tour[i];
  }

  int q;cin>>q;
  rep(i,q){
    int k;cin>>k;
    vector<pair<int,int>> X;
    rep(j,k){
      int x;cin>>x;
      X.push_back({number[x], x});
    }
    sort(all(X));
    ll ans = 0;
    rep(j,X.size()){
      int a = X[j].second;
      int b = X[(j+1)%X.size()].second;
      int p = lca(a,b);
      //cout<<a<<" "<<b<<" "<<p<<" "<<tour[first[b]-1] - tour[first[p]-1] + tour[first[a]-1] - tour[first[p]-1]<<endl;
      ans += tour[first[b]-1] - tour[first[p]-1] + tour[first[a]-1] - tour[first[p]-1];
    }
    cout<<ans/2<<endl;
  }

  return 0;
}
0