結果

問題 No.2337 Equidistant
コンテスト
ユーザー navel_tos
提出日時 2023-06-03 01:58:35
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 3,651 bytes
コンパイル時間 223 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 198,768 KB
最終ジャッジ日時 2024-12-29 02:14:25
合計ジャッジ時間 38,063 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 25 RE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#yukicoder391D Equidistant

'''
解説ACを目指す。

LCA使うところまでは良いが、部分木の頂点数を用いて計算できる点に着目できなかったのが敗因。
くやしいな。
'''
#最小共通祖先 ライブラリ
class LowestCommonAncestor:
    def __init__(self,n):
        self._n=n;n=0
        while 2**(n/10)<self._n:n+=1
        self._logn=int(n/10+2);self._depth=[0 for _ in [0]*self._n];self._distance=[0 for _ in [0]*self._n];self._ancestor=[[-1 for _ in [0]*self._n] for k in [0]*self._logn];self._edge=[[] for _ in [0]*self._n]
    def add_edge(self,u,v,w=1):self._edge[u].append((v,w));self._edge[v].append((u,w))
    def build(self,root=0):  #rootを指定し、その他の頂点に祖先情報を書き込む
        stack=[root]
        while stack:
            now=stack.pop()
            for nxt,w in self._edge[now]:
                if self._ancestor[0][nxt]!=now and self._ancestor[0][now]!=nxt:self._ancestor[0][nxt]=now;self._depth[nxt]=self._depth[now]+1;self._distance[nxt]=self._distance[now]+w;stack.append(nxt)
        for k in range(1,self._logn):
            for i in range(self._n):
                if self._ancestor[k-1][i]==-1:self._ancestor[k][i]=-1
                else:self._ancestor[k][i]=self._ancestor[k-1][self._ancestor[k-1][i]]
    def LCA(self,u,v):
        if self._depth[u]>self._depth[v]:u,v=v,u
        for k in range(self._logn-1,-1,-1):v=self._ancestor[k][v] if((self._depth[v]-self._depth[u])>>k)&1 else v
        if u==v:return u
        for k in range(self._logn-1,-1,-1):  #ギリギリ一致する直前まで祖先を辿る
            if self._ancestor[k][u]!=self._ancestor[k][v]:
                u,v=self._ancestor[k][u],self._ancestor[k][v]
        return self._ancestor[0][u]
    def distance(self,u,v):return self._distance[u]+self._distance[v]-2*self._distance[self.LCA(u,v)]


f=lambda:list(map(int,input().split()))

#入力受取
N,Q=f(); G=[[] for _ in range(N)]; LCA=LowestCommonAncestor(N)
for _ in range(N-1): a,b=f(); G[a-1].append(b-1); G[b-1].append(a-1); LCA.add_edge(a-1,b-1)

#なにかと面倒なので、親は次数1の頂点に固定
for P in range(N):
    if len(G[P])==1: break

#Vert[i]: 頂点iを親とする部分木の頂点数
Vert=[0]*N; LCA.build(P)

#DFS: 部分木の頂点数を計算
def yukicoder391D(now,back,vertice):
    vertice+=1; save=vertice; Vert[now]+=1
    for next in G[now]:
        if next==back: continue
        vertice=yukicoder391D(next,now,vertice)
        Vert[now]+=vertice-save; save=vertice
    return vertice
yukicoder391D(P,-1,0)

#タスクに回答
for _ in range(Q):
    a,b=f(); a-=1; b-=1; x=LCA.LCA(a,b)
    dx=LCA._distance[x]; da,db=LCA._distance[a]-dx,LCA._distance[b]-dx
    if (da+db)%2: print(0); continue
    if da==db:  #答えは頂点xから親方向のものすべて
        #xに入る直前まで頂点a,bを下げ、その部分木の頂点数ぶん減らした値が答え
        da-=1; bita=bin(da)[2:]; La=len(bita)
        for i in range(La):
            pos=La-1-i
            if bita[i]=='1': a=LCA._ancestor[pos][a]; b=LCA._ancestor[pos][b]
        print(N-Vert[a]-Vert[b])
    else:
        a,b,da,db=(a,b,da,db) if da<db else (b,a,db,da)  #dbが常に大きい
        #a: 中点xより親側はNG  b: 中点xより子側はNG とすればよい
        #このうち、aは部分木にはじめからカウントされていないので、bだけ下げる
        mid=(da+db)//2-1; bitm=bin(mid)[2:]; Lm=len(bitm)
        for i in range(Lm):
            pos=Lm-1-i
            if bitm[i]=='1': b=LCA._ancestor[pos][b]
        print(Vert[LCA._ancestor[0][b]]-Vert[b])
0