結果

問題 No.2337 Equidistant
ユーザー hourenhouren
提出日時 2023-06-02 23:09:40
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,106 ms / 4,000 ms
コード長 2,923 bytes
コンパイル時間 2,152 ms
コンパイル使用メモリ 175,532 KB
実行使用メモリ 51,360 KB
最終ジャッジ日時 2023-08-28 05:39:56
合計ジャッジ時間 13,511 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,376 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 3 ms
4,376 KB
testcase_07 AC 3 ms
4,376 KB
testcase_08 AC 4 ms
4,380 KB
testcase_09 AC 3 ms
4,376 KB
testcase_10 AC 3 ms
4,376 KB
testcase_11 AC 404 ms
36,216 KB
testcase_12 AC 414 ms
36,120 KB
testcase_13 AC 415 ms
36,052 KB
testcase_14 AC 412 ms
36,116 KB
testcase_15 AC 411 ms
36,120 KB
testcase_16 AC 411 ms
36,176 KB
testcase_17 AC 402 ms
36,096 KB
testcase_18 AC 414 ms
36,108 KB
testcase_19 AC 399 ms
36,184 KB
testcase_20 AC 406 ms
36,116 KB
testcase_21 AC 672 ms
51,360 KB
testcase_22 AC 303 ms
36,636 KB
testcase_23 AC 357 ms
36,724 KB
testcase_24 AC 948 ms
46,400 KB
testcase_25 AC 367 ms
36,908 KB
testcase_26 AC 1,106 ms
46,156 KB
testcase_27 AC 370 ms
36,688 KB
testcase_28 AC 365 ms
36,900 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using P = pair<ll,ll>;
#define fix(x) fixed << setprecision(x)
#define asc(x) x, vector<x>, greater<x>
#define rep(i, n) for(ll i = 0; i < n; i++)
#define all(x) (x).begin(),(x).end()
template<class T>bool chmin(T&a, const T&b){if(a>b){a=b;return 1;}return 0;}
template<class T>bool chmax(T&a, const T&b){if(a<b){a=b;return 1;}return 0;}
constexpr ll INFLL = (1LL << 62), MOD = 998244353;
constexpr int INF = (1 << 30);

struct LCA{
    int _n, _log;
    vector<int> rank;
    vector<vector<int>> d;
    LCA(vector<vector<int>>& g, int root = 0){ init(g, root); }
    void init(vector<vector<int>>& g, int root = 0){
        _n = g.size(), _log = 0;
        while((1<<_log)<=_n) _log++;
        rank = vector<int>(_n);
        rank[root] = 0;
        d = vector<vector<int>>(_n, vector<int>(_log));
        dfs(-1,root,g);
        for(int j=0;j<_log-1;j++){
            for(int i=0;i<_n;i++){
                if(d[i][j]>=0) d[i][j+1] = d[d[i][j]][j];
                else d[i][j+1] = -1;
            }
        }
    }
    void dfs(int par, int now, vector<vector<int>>& g){
        d[now][0] = par;
        for(int x:g[now]){
            if(x==par) continue;
            rank[x] = rank[now]+1;
            dfs(now,x,g);
        }
    }
    int prev(int u, int x){
        assert(x>=0);
        for(int i=0;i<_log;i++){
            if(x&1) u = d[u][i];
            x >>= 1;
            if(!x || u<0) return u;
        }
        return -1;
    }
    int lca(int u, int v){
        if(rank[u]>rank[v]) swap(u,v);
        v = prev(v, rank[v]-rank[u]);
        for(int i=_log-1;i>=0;i--){
            if(d[u][i]!=d[v][i]) u = d[u][i], v = d[v][i];
        }
        if(u!=v) return d[u][0];
        return u;
    }
    int dist(int u, int v){
        return rank[u] + rank[v] - 2*rank[lca(u,v)];
    }
};

vector<int> st;
void dfs(int par, int now, vector<vector<int>>& g){
    for(int x:g[now]){
        if(x==par) continue;
        dfs(now,x,g);
        st[now] += st[x];
    }
}

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n,q;
    cin >> n >> q;
    vector<vector<int>> g(n);
    rep(i,n-1){
        int u,v;
        cin >> u >> v;
        u--; v--;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    st = vector<int>(n,1);
    dfs(-1,0,g);
    LCA l(g);
    rep(houren,q){
        int s,t,ans;
        cin >> s >> t;
        s--; t--;
        int d = l.dist(s,t);
        if(d&1) ans = 0;
        else{
            if(l.rank[s]==l.rank[t]){
                s = l.prev(s,d/2-1);
                t = l.prev(t,d/2-1);
                ans = n - st[s] - st[t];
            }else{
                if(l.rank[s]>l.rank[t]) swap(s,t);
                s = l.prev(t,d/2-1);
                t = l.prev(t,d/2);
                ans = st[t] - st[s];
            }
        }
        cout << ans << '\n';
    }
    return 0;
}
0