結果

問題 No.898 tri-βutree
ユーザー totori_nyaatotori_nyaa
提出日時 2020-02-08 03:27:08
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 211 ms / 4,000 ms
コード長 2,002 bytes
コンパイル時間 1,971 ms
コンパイル使用メモリ 172,636 KB
実行使用メモリ 29,696 KB
最終ジャッジ日時 2024-11-08 23:24:34
合計ジャッジ時間 7,453 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 96 ms
29,696 KB
testcase_01 AC 6 ms
7,808 KB
testcase_02 AC 6 ms
7,936 KB
testcase_03 AC 6 ms
7,936 KB
testcase_04 AC 6 ms
7,936 KB
testcase_05 AC 6 ms
7,936 KB
testcase_06 AC 5 ms
8,064 KB
testcase_07 AC 190 ms
22,016 KB
testcase_08 AC 188 ms
22,144 KB
testcase_09 AC 180 ms
22,272 KB
testcase_10 AC 187 ms
22,272 KB
testcase_11 AC 192 ms
22,144 KB
testcase_12 AC 196 ms
22,144 KB
testcase_13 AC 197 ms
22,144 KB
testcase_14 AC 198 ms
22,144 KB
testcase_15 AC 196 ms
22,144 KB
testcase_16 AC 209 ms
22,016 KB
testcase_17 AC 187 ms
22,144 KB
testcase_18 AC 195 ms
22,272 KB
testcase_19 AC 181 ms
22,144 KB
testcase_20 AC 211 ms
22,272 KB
testcase_21 AC 206 ms
22,144 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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


// LCA
// verify: https://atcoder.jp/contests/abc014/submissions/9535352
vector<vector<array<ll,2>>> v(200010);
bool used[200010];
int depth[200010];
int par[200010][20];
ll dp[200020];

// 根からの深さと1つ上の親を求める. initの中で使う
void dfs(int p,int d,ll cost){
    used[p] = 1;
    depth[p] = d;
    dp[p]=cost;
    for(auto i:v[p]){
        if(!used[i[0]]){
            dfs(i[0],d+1,cost+i[1]);
            par[i[0]][0] = p;
        }
    }
}

// init(n, root) でダブリングテーブルを作製する, nは頂点数
void init(int n,int root=0){
    for(int i=0;i<n;i++){
        used[i]=false;
        for(int j=0;j<20;j++){
            par[i][j] = -1;
        }
    }
    dfs(root,0,0);
    for(int j=1;j<20;j++){
        for(int i=0;i<n;i++){
            if(par[i][j-1] == -1) continue;
            par[i][j] = par[par[i][j-1]][j-1];
        }
    }
}

//頂点aのd個上の親を求める
ll get_nyaa(ll a,ll d){
    for(ll i=19;i>=0;i--){
        if(d & (1<<i)) a = par[a][i];
    }
    return a;
}

// a,bのLCAを求める
int lca(int a,int b){
    if(depth[a] < depth[b]) swap(a,b);
    int dif = depth[a] - depth[b];
    for(int i=19;i>=0;i--){
        if(dif & (1<<i)) a = par[a][i];
    }
    if(a==b) return a;
    for(int i=19;i>=0;i--){
        if(par[a][i]!=par[b][i]){
            a = par[a][i], b = par[b][i];
        }
    }
    return par[a][0];
}

signed main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    cout << fixed << setprecision(20);
 
    ll n;
    cin>>n;
    for(int i=1;i<n;i++){
        ll a,b,c;
        cin>>a>>b>>c;
        v[b].push_back({a,c});
        v[a].push_back({b,c});
    }
    init(n);
    int q; cin>>q;
    while(q--){
        int a,b,c;
        cin>>a>>b>>c;
        ll ans = dp[a] + dp[b] + dp[c];
        ans -= dp[lca(a,b)] + dp[lca(a,c)] + dp[lca(b,c)];
        // ans += dp[lca(a,lca(b,c))];
        cout << ans << "\n";
    }

    
}
0