結果

問題 No.2337 Equidistant
ユーザー navel_tosnavel_tos
提出日時 2023-06-03 01:58:35
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 3,651 bytes
コンパイル時間 277 ms
コンパイル使用メモリ 82,292 KB
実行使用メモリ 198,852 KB
最終ジャッジ日時 2024-06-09 03:07:39
合計ジャッジ時間 32,845 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
53,724 KB
testcase_01 AC 34 ms
53,716 KB
testcase_02 AC 32 ms
54,772 KB
testcase_03 AC 32 ms
53,768 KB
testcase_04 AC 32 ms
54,620 KB
testcase_05 AC 31 ms
53,728 KB
testcase_06 AC 144 ms
78,996 KB
testcase_07 AC 156 ms
79,548 KB
testcase_08 AC 179 ms
80,016 KB
testcase_09 AC 143 ms
78,628 KB
testcase_10 AC 149 ms
79,228 KB
testcase_11 AC 1,904 ms
169,396 KB
testcase_12 AC 1,656 ms
167,964 KB
testcase_13 AC 1,657 ms
168,224 KB
testcase_14 AC 1,884 ms
168,048 KB
testcase_15 AC 1,896 ms
167,832 KB
testcase_16 AC 1,681 ms
167,732 KB
testcase_17 AC 1,967 ms
169,156 KB
testcase_18 AC 1,625 ms
168,372 KB
testcase_19 AC 1,651 ms
168,364 KB
testcase_20 AC 1,945 ms
168,980 KB
testcase_21 RE -
testcase_22 AC 1,458 ms
198,852 KB
testcase_23 AC 1,818 ms
174,236 KB
testcase_24 RE -
testcase_25 AC 2,018 ms
175,392 KB
testcase_26 RE -
testcase_27 AC 2,075 ms
175,508 KB
testcase_28 AC 2,053 ms
175,284 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)]


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