結果

問題 No.2337 Equidistant
コンテスト
ユーザー houren
提出日時 2023-06-02 22:53:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,903 bytes
コンパイル時間 3,270 ms
コンパイル使用メモリ 176,536 KB
実行使用メモリ 51,712 KB
最終ジャッジ日時 2024-12-28 21:08:45
合計ジャッジ時間 18,957 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 6 WA * 1 RE * 21
権限があれば一括ダウンロードができます

ソースコード

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) 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*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.assign(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