結果

問題 No.901 K-ary εxtrεεmε
ユーザー Taiki0715Taiki0715
提出日時 2024-04-30 20:14:56
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 321 ms / 3,000 ms
コード長 5,870 bytes
コンパイル時間 6,383 ms
コンパイル使用メモリ 327,920 KB
実行使用メモリ 34,304 KB
最終ジャッジ日時 2024-11-20 17:55:43
合計ジャッジ時間 14,136 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 126 ms
34,304 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 4 ms
5,248 KB
testcase_03 AC 4 ms
5,248 KB
testcase_04 AC 4 ms
5,248 KB
testcase_05 AC 4 ms
5,248 KB
testcase_06 AC 3 ms
5,248 KB
testcase_07 AC 244 ms
28,032 KB
testcase_08 AC 257 ms
28,000 KB
testcase_09 AC 256 ms
28,104 KB
testcase_10 AC 247 ms
28,160 KB
testcase_11 AC 252 ms
27,992 KB
testcase_12 AC 247 ms
28,032 KB
testcase_13 AC 236 ms
28,000 KB
testcase_14 AC 235 ms
28,032 KB
testcase_15 AC 238 ms
28,088 KB
testcase_16 AC 243 ms
27,988 KB
testcase_17 AC 317 ms
28,012 KB
testcase_18 AC 321 ms
28,056 KB
testcase_19 AC 316 ms
28,032 KB
testcase_20 AC 316 ms
28,032 KB
testcase_21 AC 314 ms
28,112 KB
testcase_22 AC 241 ms
28,400 KB
testcase_23 AC 234 ms
28,416 KB
testcase_24 AC 236 ms
28,312 KB
testcase_25 AC 242 ms
28,544 KB
testcase_26 AC 244 ms
28,424 KB
testcase_27 AC 320 ms
26,068 KB
testcase_28 AC 311 ms
25,984 KB
testcase_29 AC 316 ms
25,984 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
template<int mod>istream &operator>>(istream &is,static_modint<mod> &a){long long b;is>>b;a=b;return is;}
istream &operator>>(istream &is,modint &a){long long b;cin>>b;a=b;return is;}
#endif
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) static_cast<void>(0)
#define debugg(...) static_cast<void>(0)
#define esper(...) static_cast<void>(0)
template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<' '<<p.second;return os;}
#endif
using ll=long long;
using ull=unsigned long long;
using P=pair<ll,ll>;
template<typename T>using minque=priority_queue<T,vector<T>,greater<T>>;
template<typename T>bool chmax(T &a,const T &b){return (a<b?(a=b,true):false);}
template<typename T>bool chmin(T &a,const T &b){return (a>b?(a=b,true):false);}
template<typename T1,typename T2>istream &operator>>(istream &is,pair<T1,T2>&p){is>>p.first>>p.second;return is;}
template<typename T>istream &operator>>(istream &is,vector<T> &a){for(auto &i:a)is>>i;return is;}
template<typename T1,typename T2>void operator++(pair<T1,T2>&a,int n){a.first++,a.second++;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int n){a.first--,a.second--;}
template<typename T>void operator++(vector<T>&a,int n){for(auto &i:a)i++;}
template<typename T>void operator--(vector<T>&a,int n){for(auto &i:a)i--;}
#define reps(i,a,n) for(int i=(a);i<(n);i++)
#define rep(i,n) reps(i,0,n)
#define all(x) x.begin(),x.end()
#define pcnt(x) __builtin_popcountll(x)
#define fin(x) return cout<<x<<'\n',static_cast<void>(0)
ll myceil(ll a,ll b){return (a+b-1)/b;}
template<typename T,size_t n,size_t id=0>
auto vec(const int (&d)[n],const T &init=T()){
  if constexpr (id<n)return vector(d[id],vec<T,n,id+1>(d,init));
  else return init;
}
void SOLVE();
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cout<<fixed<<setprecision(16);
  #ifdef LOCAL
  clock_t start=clock();
  #endif
  int testcase=1;
  //cin>>testcase;
  for(int i=0;i<testcase;i++){
    SOLVE();
  }
  #ifdef LOCAL
  cerr<<"time:";
  cerr<<(clock()-start)/1000;
  cerr<<"ms\n";
  #endif
}
struct HeavyLightDecomposition{
  int n;
  vector<vector<int>>to;
  int root;
  vector<int>par,pathtop,in,out,dep;
  vector<int>sub;
private:
  void st_dfs(int x,int p){
    par[x]=p;
    sub[x]=1;
    for(auto &i:to[x]){
      if(i==p){
        if(i==to[x].back())break;
        else swap(i,to[x].back());
      }
      st_dfs(i,x);
      sub[x]+=sub[i];
      if(sub[i]>sub[to[x][0]]){
        swap(i,to[x][0]);
      }
    }
  }
  void hld_dfs(int x,int p,int &id){
    in[x]=id++;
    for(auto i:to[x])if(i!=p){
      dep[i]=dep[x]+1;
      pathtop[i]=(i==to[x][0]?pathtop[x]:i);
      hld_dfs(i,x,id);
    }
    out[x]=id;
  }
public:
  HeavyLightDecomposition(int n,int root=0):n(n),to(n),sub(n),par(n),in(n),out(n),root(root),pathtop(n),dep(n),g(n){}
  void add_edge(int u,int v){
    to[u].emplace_back(v);
    to[v].emplace_back(u);
  }
  void build(){
    dep[root]=0;
    st_dfs(root,-1);
    pathtop[root]=root;
    int id=0;
    hld_dfs(root,-1,id);
  }
  inline int get(int x)const{return in[x];}
  int lca(int u,int v)const{
    int pu=pathtop[u],pv=pathtop[v];
    while(pathtop[u]!=pathtop[v]){
      if(in[pu]>in[pv])u=par[pu],pu=pathtop[u];
      else v=par[pv],pv=pathtop[v];
    }
    return (in[u]>in[v]?v:u);
  }
  void subtree_query(int u ,const auto&f){
    static_assert(is_convertible_v<decltype(f),function<void(int,int)>>);
    f(in[u],out[u]);
  }
  void path_query(int u,int v,const auto&f){
    static_assert(is_convertible_v<decltype(f),function<void(int,int)>>);
    int pu=pathtop[u],pv=pathtop[v];
    while(pathtop[u]!=pathtop[v]){
      if(in[u]>in[v])f(in[pu],in[u]+1),u=par[pu],pu=pathtop[u];
      else f(in[pv],in[v]+1),v=par[pv],pv=pathtop[v];
    }
    if(in[u]>in[v])swap(u,v);
    f(in[u],in[v]+1);
  }
  void noncommutative_path_query(int u,int v,const auto&f){
    static_assert(is_convertible_v<decltype(f),function<void(int,int)>>);
    int l=lca(u,v);
    while(pathtop[u]!=pathtop[l]){
      f(in[u]+1,in[pathtop[u]]);
      u=par[pathtop[u]];
    }
    if(u!=l)f(in[u]+1,in[l]+1);
    f(in[l],in[l]+1);
    if(v==l)return;
    vector<pair<int,int>>query;
    while(true){
      if(pathtop[l]==pathtop[v]){
        query.emplace_back(in[l]+1,in[v]+1);
        break;
      }
      query.emplace_back(in[pathtop[v]],in[v]+1);
      v=par[pathtop[v]];
    }
    reverse(all(query));
    for(auto [i,j]:query)f(i,j);
  }
  vector<vector<int>>g;
  int auxiliary_tree(vector<int>v){
    sort(all(v),[&](int x,int y){return in[x]<in[y];});
    v.reserve(v.size()*2-1);
    int vs=v.size();
    reps(i,1,vs)v.push_back(lca(v[i-1],v[i]));
    sort(all(v),[&](int x,int y){return in[x]<in[y];});
    v.erase(unique(all(v)),v.end());
    rep(i,v.size())g[v[i]].clear();
    stack<int>st;
    rep(i,v.size()){
      while(!st.empty()&&out[st.top()]<=in[v[i]])st.pop();
      if(!st.empty()){
        g[st.top()].push_back(v[i]);
        g[v[i]].push_back(st.top());
      }
      st.push(v[i]);
    }
    while(st.size()>1)st.pop();
    return st.top();
  }
};
void SOLVE(){
  int n;
  cin>>n;
  HeavyLightDecomposition hld(n*2-1);
  vector<ll>w(n-1);
  rep(i,n-1){
    int u,v;
    cin>>u>>v>>w[i];
    hld.add_edge(i+n,u);
    hld.add_edge(i+n,v);
  }
  hld.build();
  fenwick_tree<ll>BIT(n*2-1);
  rep(i,n-1)BIT.add(hld.get(i+n),w[i]);
  int q;
  cin>>q;
  while(q--){
    int k;
    cin>>k;
    vector<int>x(k);
    cin>>x;
    int root=hld.auxiliary_tree(x);
    ll ans=0;
    auto dfs=[&](auto self,int x,int p)->void {
      for(auto i:hld.g[x])if(i!=p){
        hld.path_query(x,i,[&](int l,int r){ans+=BIT.sum(l,r);});
        self(self,i,x);
      }
    };
    dfs(dfs,root,-1);
    cout<<ans<<endl;
  }
}
0