結果

問題 No.2337 Equidistant
ユーザー navel_tosnavel_tos
提出日時 2023-06-03 02:00:20
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,063 ms / 4,000 ms
コード長 3,719 bytes
コンパイル時間 319 ms
コンパイル使用メモリ 86,896 KB
実行使用メモリ 349,844 KB
最終ジャッジ日時 2023-08-28 07:25:23
合計ジャッジ時間 32,293 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 71 ms
71,616 KB
testcase_01 AC 69 ms
71,752 KB
testcase_02 AC 71 ms
71,708 KB
testcase_03 AC 70 ms
71,724 KB
testcase_04 AC 70 ms
71,504 KB
testcase_05 AC 69 ms
71,456 KB
testcase_06 AC 163 ms
80,992 KB
testcase_07 AC 163 ms
80,752 KB
testcase_08 AC 163 ms
80,804 KB
testcase_09 AC 164 ms
79,864 KB
testcase_10 AC 163 ms
81,052 KB
testcase_11 AC 1,612 ms
171,768 KB
testcase_12 AC 1,360 ms
171,984 KB
testcase_13 AC 1,355 ms
171,008 KB
testcase_14 AC 1,597 ms
170,904 KB
testcase_15 AC 1,610 ms
170,944 KB
testcase_16 AC 1,348 ms
170,904 KB
testcase_17 AC 1,619 ms
172,192 KB
testcase_18 AC 1,327 ms
171,112 KB
testcase_19 AC 1,408 ms
171,708 KB
testcase_20 AC 1,581 ms
171,280 KB
testcase_21 AC 1,678 ms
345,996 KB
testcase_22 AC 1,129 ms
198,460 KB
testcase_23 AC 1,510 ms
177,168 KB
testcase_24 AC 1,891 ms
346,268 KB
testcase_25 AC 1,726 ms
177,616 KB
testcase_26 AC 2,063 ms
349,844 KB
testcase_27 AC 1,623 ms
177,108 KB
testcase_28 AC 1,612 ms
176,984 KB
権限があれば一括ダウンロードができます

ソースコード

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)]


import sys; sys.setrecursionlimit(10**7); input=sys.stdin.readline
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