結果

問題 No.2337 Equidistant
ユーザー gucci0512gucci0512
提出日時 2023-06-02 22:19:31
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,982 bytes
コンパイル時間 4,372 ms
コンパイル使用メモリ 245,668 KB
実行使用メモリ 95,364 KB
最終ジャッジ日時 2023-08-28 04:03:01
合計ジャッジ時間 24,326 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
17,528 KB
testcase_01 WA -
testcase_02 AC 8 ms
17,512 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 AC 1,298 ms
72,208 KB
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
// url 
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=b-1;i>=a;i--)
#define all(x) (x).begin(),(x).end()
#define pb(x) push_back(x);
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a = b; return 1; } return 0; }
typedef long long ll;
typedef long double lld;
using namespace std;
using namespace atcoder;
using mint = static_modint<998244353>;
const ll mod=998244353;
// using mint = static_modint<1000000007>;
//const ll mod=1e9+7;
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
const string zton="0123456789";
const string atoz="abcdefghijklmnopqrstuvwxyz";
const string ATOZ="ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const ll inf=(1ll<<60);
// const int inf=(1<<30);
lld dist(lld x1,lld x2,lld y1,lld y2){
    lld res=(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
    res=sqrt(abs(res));
    return res;
}
lld arg(lld x,lld y){
    const lld eps=1e-8;
    lld res=0;
    if(abs(x)+abs(y)<=eps)return 0.0;
    else if(abs(x)<=eps){
        if(y>=0.0)return (M_PI/2);
        else return (M_PI/2+M_PI);
    }
    else if(abs(y)<=eps){
        if(x>=0.0)return 0.0;
        else return M_PI;
    }
    res=atan2(abs(y),abs(x));
    if(x<=0&&y>=0)res=(M_PI-res);
    else if(x<=0&&y<=0)res+=(M_PI);
    else if(x>=0&&y<=0)res=(M_PI*2-res);
    return res;
}
ll gcd(ll a,ll b){
    if(a==0||b==0)return a+b;
    ll r;
    r=a%b;
    if(r==0){
        return b;
    }
    else{
        return gcd(b,r);
    }
}
typedef pair<ll,int> P;
struct Edge {
    long long to;
};
using Graph = vector<vector<Edge>>;
/* LCA(G, root): 木 G に対する根を root として Lowest Common Ancestor を求める構造体
    query(u,v): u と v の LCA を求める。計算量 O(logn)
    前処理: O(nlogn)時間, O(nlogn)空間
*/
struct LCA {
    vector<vector<int>> parent;  // parent[k][u]:= u の 2^k 先の親
    vector<int> dist;            // root からの距離
    LCA(const Graph &G, int root = 0) { init(G, root); }
    // 初期化
    void init(const Graph &G, int root = 0) {
        int V = G.size();
        int K = 1;
        while ((1 << K) < V) K++;
        parent.assign(K, vector<int>(V, -1));
        dist.assign(V, -1);
        dfs(G, root, -1, 0);
        for (int k = 0; k + 1 < K; k++) {
            for (int v = 0; v < V; v++) {
                if (parent[k][v] < 0) {
                    parent[k + 1][v] = -1;
                } else {
                    parent[k + 1][v] = parent[k][parent[k][v]];
                }
            }
        }
    }
    // 根からの距離と1つ先の頂点を求める
    void dfs(const Graph &G, int v, int p, int d) {
        parent[0][v] = p;
        dist[v] = d;
        for (auto e : G[v]) {
            if (e.to != p) dfs(G, e.to, v, d + 1);
        }
    }
    int query(int u, int v) {
        if (dist[u] < dist[v]) swap(u, v);  // u の方が深いとする
        int K = parent.size();
        // LCA までの距離を同じにする
        for (int k = 0; k < K; k++) {
            if ((dist[u] - dist[v]) >> k & 1) {
                u = parent[k][u];
            }
        }
        // 二分探索で LCA を求める
        if (u == v) return u;
        for (int k = K - 1; k >= 0; k--) {
            if (parent[k][u] != parent[k][v]) {
                u = parent[k][u];
                v = parent[k][v];
            }
        }
        return parent[0][u];
    }
    int get_post(int v,int u){
        int K=parent.size();
        int distu=dist[u];
        int distv=dist[v]+1;
        for(int k=K-1;k>=0;k--){
            if(distu-distv>=(1<<k)){
                u=parent[k][u];
            }
        }
        return u;
    }
    int get_half(int s,int t){
        int u=query(s,t);
        int K=parent.size();
        int target_dist=dist[t]-get_dist(s,t)/2;
        for(int k=K-1;k>=0;k--){
            if(dist[t]-target_dist>=(1<<k)){
                t=parent[k][t];
            }
        }
        return t;
    }
    int get_dist(int u, int v) { return dist[u] + dist[v] - 2 * dist[query(u, v)]; }
};
    // int n;cin >> n;
    // Graph g(n);
    // rep(i,0,n){
    //     int k;cin >> k;
    //     rep(j,0,k){
    //         Edge c;
    //         cin >> c.to;
    //         g[i].push_back(c);
    //     }
    // }
    // LCA aa=LCA(g,0);
    // int q;cin >> q;
    // rep(i,0,q){
    //     int u,v;cin >> u >> v;
    //     cout << aa.query(u,v) << endl;
    // }
vector<int> G[201010];
map<int,int> F[201010];

int dfs(int cu,int pa){
    F[cu][cu]=1;
    for(int to:G[cu]){
        if(to==pa)continue;
        F[cu][to]=dfs(to,cu);
        F[cu][cu]+=F[cu][to];
    }
    return F[cu][cu];
}

int main(void){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int N,Q;cin >> N >> Q;
    
    Graph g(N);
    rep(i,0,N-1){
        int a,b;cin >> a >> b;
        a--;b--;
        Edge c;
        c.to=b;
        g[a].push_back(c);
        c.to=a;
        g[b].push_back(c);
        G[a].pb(b);
        G[b].pb(a);
    }
    dfs(0,-1);
    LCA lca=LCA(g,0);
    vector<int> ans;
    rep(i,0,Q){
        int s,t;cin >> s >> t;s--;t--;
        int dists=lca.get_dist(s,0);
        int distt=lca.get_dist(t,0);
        if(dists==distt){
            int u=lca.query(s,t);
            int s2=lca.get_post(u,s);
            int t2=lca.get_post(u,t);
            ans.pb(F[u][u]-F[u][s2]-F[u][t2]);
        }
        else{
            int distst=lca.get_dist(s,t);
            if(distst%2==1){
                ans.pb(0);
            }
            else{
                if(dists>distt){
                    swap(s,t);
                    swap(dists,distt);
                }
                int u=lca.get_half(s,t);
                int t2=lca.get_post(u,t);
                ans.pb(F[u][u]-F[u][t2]);
            }
        }
    }
    for(int x:ans){
        cout << x << endl;
    }
    // rep(i,0,N){
    //     cout << F[i][i] << endl;
    // }
}
0