結果
問題 | No.901 K-ary εxtrεεmε |
ユーザー | smiken_61 |
提出日時 | 2019-11-02 20:46:10 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 479 ms / 3,000 ms |
コード長 | 11,808 bytes |
コンパイル時間 | 2,003 ms |
コンパイル使用メモリ | 172,180 KB |
実行使用メモリ | 46,336 KB |
最終ジャッジ日時 | 2024-09-14 23:16:36 |
合計ジャッジ時間 | 15,260 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 207 ms
46,336 KB |
testcase_01 | AC | 6 ms
8,192 KB |
testcase_02 | AC | 7 ms
8,192 KB |
testcase_03 | AC | 7 ms
8,064 KB |
testcase_04 | AC | 7 ms
8,192 KB |
testcase_05 | AC | 7 ms
8,064 KB |
testcase_06 | AC | 7 ms
8,192 KB |
testcase_07 | AC | 459 ms
40,704 KB |
testcase_08 | AC | 450 ms
40,832 KB |
testcase_09 | AC | 462 ms
40,832 KB |
testcase_10 | AC | 474 ms
40,832 KB |
testcase_11 | AC | 479 ms
40,832 KB |
testcase_12 | AC | 462 ms
40,832 KB |
testcase_13 | AC | 461 ms
40,832 KB |
testcase_14 | AC | 465 ms
40,832 KB |
testcase_15 | AC | 465 ms
40,832 KB |
testcase_16 | AC | 456 ms
40,832 KB |
testcase_17 | AC | 414 ms
40,832 KB |
testcase_18 | AC | 409 ms
40,832 KB |
testcase_19 | AC | 403 ms
40,704 KB |
testcase_20 | AC | 409 ms
40,832 KB |
testcase_21 | AC | 403 ms
40,832 KB |
testcase_22 | AC | 442 ms
40,960 KB |
testcase_23 | AC | 423 ms
40,832 KB |
testcase_24 | AC | 424 ms
40,960 KB |
testcase_25 | AC | 422 ms
40,960 KB |
testcase_26 | AC | 437 ms
40,832 KB |
testcase_27 | AC | 276 ms
40,704 KB |
testcase_28 | AC | 268 ms
40,704 KB |
testcase_29 | AC | 271 ms
40,832 KB |
ソースコード
#include <bits/stdc++.h> // #include <boost/multiprecision/cpp_int.hpp> #define int long long #define inf 1000000007 #define pa pair<int,int> #define ll long long #define pal pair<double,double> #define ppap pair<pa,int> #define PI 3.14159265358979323846 #define paa pair<int,char> #define mp make_pair #define pb push_back #define EPS (1e-8) int dx[8]={0,1,0,-1,1,1,-1,-1}; int dy[8]={1,0,-1,0,-1,1,1,-1}; using namespace std; class pa3{ public: int x; int y,z; pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {} bool operator < (const pa3 &p) const{ if(x!=p.x) return x<p.x; if(y!=p.y) return y<p.y; return z<p.z; //return x != p.x ? x<p.x: y<p.y; } bool operator > (const pa3 &p) const{ if(x!=p.x) return x>p.x; if(y!=p.y) return y>p.y; return z>p.z; //return x != p.x ? x<p.x: y<p.y; } bool operator == (const pa3 &p) const{ return x==p.x && y==p.y && z==p.z; } bool operator != (const pa3 &p) const{ return !( x==p.x && y==p.y && z==p.z); } }; class pa4{ public: int x; int y,z,w; pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {} bool operator < (const pa4 &p) const{ if(x!=p.x) return x<p.x; if(y!=p.y) return y<p.y; if(z!=p.z)return z<p.z; return w<p.w; //return x != p.x ? x<p.x: y<p.y; } bool operator > (const pa4 &p) const{ if(x!=p.x) return x>p.x; if(y!=p.y) return y>p.y; if(z!=p.z)return z>p.z; return w>p.w; //return x != p.x ? x<p.x: y<p.y; } bool operator == (const pa4 &p) const{ return x==p.x && y==p.y && z==p.z &&w==p.w; } }; class pa2{ public: int x,y; pa2(int x=0,int y=0):x(x),y(y) {} pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);} pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);} bool operator < (const pa2 &p) const{ return y != p.y ? y<p.y: x<p.x; } bool operator > (const pa2 &p) const{ return x != p.x ? x<p.x: y<p.y; } bool operator == (const pa2 &p) const{ return abs(x-p.x)==0 && abs(y-p.y)==0; } bool operator != (const pa2 &p) const{ return !(abs(x-p.x)==0 && abs(y-p.y)==0); } }; string itos( int i ) { ostringstream s ; s << i ; return s.str() ; } int gcd(int v,int b){ if(v==0) return b; if(b==0) return v; if(v>b) return gcd(b,v); if(v==b) return b; if(b%v==0) return v; return gcd(v,b%v); } int mod; int extgcd(int a, int b, int &x, int &y) { if (b == 0) { x = 1; y = 0; return a; } int d = extgcd(b, a%b, y, x); y -= a/b * x; return d; } pa operator+(const pa & l,const pa & r) { return {l.first+r.first,l.second+r.second}; } pa operator-(const pa & l,const pa & r) { return {l.first-r.first,l.second-r.second}; } int pr[1000010]; int inv[1000010]; int beki(int wa,int rr,int warukazu){ if(rr==0) return 1%warukazu; if(rr==1) return wa%warukazu; wa%=warukazu; if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu; ll zx=beki(wa,rr/2,warukazu); return (zx*zx)%warukazu; } int comb(int nn,int rr){ if(rr<0 || rr>nn || nn<0) return 0; int r=pr[nn]*inv[rr]; r%=mod; r*=inv[nn-rr]; r%=mod; return r; } void gya(int ert){ pr[0]=1; for(int i=1;i<=ert;i++){ pr[i]=(pr[i-1]*i)%mod; } inv[ert]=beki(pr[ert],mod-2,mod); for(int i=ert-1;i>=0;i--){ inv[i]=inv[i+1]*(i+1)%mod; } } // cin.tie(0); // ios::sync_with_stdio(false); //priority_queue<pa3,vector<pa3>,greater<pa3>> pq; //sort(ve.begin(),ve.end(),greater<int>()); // mt19937(clock_per_sec); // mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()) ; //----------------kokomade tenpure------------ // 範囲を一様にa倍してb足す操作.区間maxminsumに対応 //要はupdate, addにも対応して3種の区間クエリに回答可能 using Node =int; using Sayo =int; struct seg_chusho{ // 1 // 2 3 // 4 5 6 7 private: public: // (1<<15)=32768 // 1<<16 = 65536 // 1<<17 = 131072 // 1<<18 = 262144 int cor=(1<<18); /* struct Node{ int mi; int ma; int su; }; struct Sayo{ int a; int b; bool operator==(sayo v){ return a==v.a && b==v.b; } }; */ // 遅延が貯まっていないときに保持する値 Sayo S_id=-1; Node N_id=0; Node node_merge(Node x,Node y){ // x と yをマージする 区間和なら和 return x+y; // ここに実装する } Sayo merge_sayo(Sayo x,Sayo y){ // 作用xの上から作用yを更にのっける if(y==-1) return x; return y; // ここに実装する } Node node_sayo(Node x,Sayo y,int l,int r,int k){ // 区間[l,r)のNode x に作用yを載せる // x == N_idの心配は基本的に不要 // y == S_id も不必要 return y*(r-l); // ここに実装する } vector<Node> vec; vector<Sayo> lazy; void shoki1(){ //S_id,N_idが未定義ならここでやる vec.resize(2*cor+3, N_id); lazy.resize(2*cor+3,S_id); } void shoki2(){ for(int i=cor-1;i>0;i--) vec[i]=node_merge(vec[2*i],vec[2*i+1]); } void chien(int k,int l,int r){ if(lazy[k]==S_id) return; if(l+1<r){ vec[k]=node_sayo(vec[k],lazy[k],l,r,k); lazy[2*k]=merge_sayo(lazy[2*k],lazy[k]); lazy[2*k+1]=merge_sayo(lazy[2*k+1],lazy[k]); lazy[k]=S_id; } else { vec[k]=node_sayo(vec[k],lazy[k],l,r,k); lazy[k]=S_id; } } Node rangeshori(int a,int b,Sayo w, int k=1,int l=0,int r=-3){ if(r<0) r=cor; if(a<=l && r<=b){ lazy[k]=merge_sayo(lazy[k],w); chien(k,l,r); return vec[k]; } if(r<=a || b<=l){ chien(k,l,r); return vec[k]; } chien(k,l,r); Node w1=rangeshori(a,b,w,k*2,l,(l+r)/2); Node w2=rangeshori(a,b,w,k*2+1,(l+r)/2,r); vec[k]=node_merge(w1,w2); return vec[k]; } // [a,b) // k-th node // k no kukanha [l,r) Node getrange(int a,int b,int k=1,int l=0,int r=-1){ if(r<0) r=cor; chien(k,l,r); if(r<=a || b<=l){// 区間外 return N_id; } if(a<=l && r<=b){// 完全含む return vec[k]; } Node w1=getrange(a,b,k*2,l,(l+r)/2); Node w2=getrange(a,b,k*2+1,(l+r)/2,r); return node_merge(w1,w2); } }; seg_chusho ss; vector<pa> G[100020]; vector<int> G2[100020]; int depth[100020],val[100020]; int cnt=0; int iti[100020]; int dp[33][100020]; void dfs(int r,int p,int d,int t){ depth[r]=d; iti[r]=cnt; dp[0][r]=p; cnt++; val[r]=t; for(auto v:G[r])if(v.first!=p){ dfs(v.first,r,d+1,t+v.second); G2[r].pb(v.first); } } int lca(int a,int b){ if(depth[a]<depth[b])swap(a,b); int sa=depth[a]-depth[b]; for(int i=0;i<30;i++)if(sa&(1<<i))a=dp[i][a]; if(a==b) return a; for(int i=29;i>=0;i--)if(dp[i][a]!=dp[i][b]){ a=dp[i][a],b=dp[i][b]; } return dp[0][a]; } int dis(int a,int b){ return val[a]+val[b]-2*val[lca(a,b)]; } bool hikaku(int a,int b){ return iti[a]<iti[b]; } signed main(){ cin.tie(0); ios::sync_with_stdio(false); int n; cin>>n; for(int i=0;i<n-1;i++){ int y,yy,yyy; cin>>y>>yy>>yyy; G[y].pb(mp(yy,yyy)); G[yy].pb(mp(y,yyy)); } dfs(0,0,0,0); for(int i=1;i<30;i++)for(int j=0;j<n;j++)dp[i][j]=dp[i-1][dp[i-1][j]]; int q; cin>>q; for(int rt=0;rt<q;rt++){ int t; cin>>t; int ans=0; vector<int> ve(t); for(int i=0;i<t;i++)cin>>ve[i]; sort(ve.begin(),ve.end(),hikaku); for(int i=0;i<t-1;i++)ans+=dis(ve[i],ve[i+1]); ans+=dis(ve[0],ve.back()); cout<<ans/2<<endl; } return 0; }