結果

問題 No.1094 木登り / Climbing tree
ユーザー KKT89KKT89
提出日時 2020-06-27 10:51:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 826 ms / 2,000 ms
コード長 1,938 bytes
コンパイル時間 1,418 ms
コンパイル使用メモリ 97,332 KB
実行使用メモリ 46,652 KB
最終ジャッジ日時 2024-11-08 06:11:02
合計ジャッジ時間 21,140 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 6 ms
8,064 KB
testcase_01 AC 787 ms
42,044 KB
testcase_02 AC 129 ms
46,652 KB
testcase_03 AC 176 ms
9,452 KB
testcase_04 AC 177 ms
22,304 KB
testcase_05 AC 297 ms
37,412 KB
testcase_06 AC 379 ms
18,028 KB
testcase_07 AC 802 ms
41,916 KB
testcase_08 AC 800 ms
41,916 KB
testcase_09 AC 801 ms
41,784 KB
testcase_10 AC 826 ms
41,788 KB
testcase_11 AC 817 ms
41,792 KB
testcase_12 AC 787 ms
41,916 KB
testcase_13 AC 793 ms
41,912 KB
testcase_14 AC 788 ms
41,912 KB
testcase_15 AC 405 ms
16,128 KB
testcase_16 AC 611 ms
37,348 KB
testcase_17 AC 501 ms
25,092 KB
testcase_18 AC 469 ms
20,856 KB
testcase_19 AC 545 ms
32,004 KB
testcase_20 AC 801 ms
41,916 KB
testcase_21 AC 511 ms
26,452 KB
testcase_22 AC 801 ms
41,916 KB
testcase_23 AC 804 ms
41,912 KB
testcase_24 AC 780 ms
42,040 KB
testcase_25 AC 783 ms
41,916 KB
testcase_26 AC 795 ms
41,916 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <queue>
#include <cstdio>
#include <set>
using namespace std;
typedef long long int ll;

struct LowestCommonAncester{
    int n,h;
    vector<vector<int>> g,par;
    vector<int> dep;
    LowestCommonAncester(int n):n(n),g(n),dep(n){
        h=1;
        while((1<<h)<=n)h++;
        par.assign(h,vector<int> (n,-1));
    }
    void add_edge(int u,int v){
        g[u].push_back(v);
        g[v].push_back(u);
    }
    void dfs(int v,int p,int d){
        par[0][v]=p;
        dep[v]=d;
        for(auto u:g[v]){
            if(u!=p)dfs(u,v,d+1);
        }
    }
    void build(int r){
        dfs(r,-1,0);
        for(int i=0;i<h-1;i++){
            for(int j=0;j<n;j++){
                if(par[i][j]>=0)par[i+1][j]=par[i][par[i][j]];
            }
        }
    }
    int lca(int u,int v){
        if(dep[u]>dep[v])swap(u,v);
        for(int i=0;i<h;i++){
            if((dep[v]-dep[u])&1<<i)v=par[i][v];
        }
        if(u==v)return u;
        for(int i=h-1;i>=0;i--){
            if(par[i][u]!=par[i][v])u=par[i][u],v=par[i][v];
        }
        return par[0][u];
    }
    int distance(int u,int v){
        return dep[u]+dep[v]-dep[lca(u,v)]*2;
    }
};

vector<pair<int,int>> g[200200];
int d[200200];

void dfs(int s,int p){
    for(auto t:g[s]){
        if(t.first==p)continue;
        d[t.first]=d[s]+t.second;
        dfs(t.first,s);
    }
}

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n; cin >> n;
    LowestCommonAncester LCA(n);
    for(int i=1;i<n;i++){
        int a,b,c; cin >> a >> b >> c;
        a--; b--;
        g[a].push_back({b,c});
        g[b].push_back({a,c});
        LCA.add_edge(a,b);
    }
    LCA.build(0);
    dfs(0,-1);
    int q; cin >> q;
    while(q--){
        int s,t; cin >> s >> t;
        s--; t--;
        int l=LCA.lca(s,t);
        cout << d[s]+d[t]-2*d[l] << endl;
    }
}
0