結果

問題 No.1424 Ultrapalindrome
ユーザー 👑 SPD_9X2
提出日時 2021-03-12 21:33:58
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,098 bytes
コンパイル時間 305 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 101,052 KB
最終ジャッジ日時 2024-10-14 11:44:20
合計ジャッジ時間 5,247 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff #

"""

全ての葉の間の距離がひとしければおk

"""

import sys
from sys import stdin

from collections import deque
def NC_Dij(lis,start):

    ret = [float("inf")] * len(lis)
    ret[start] = 0
    
    q = deque([start])
    plis = [i for i in range(len(lis))]

    while len(q) > 0:
        now = q.popleft()

        for nex in lis[now]:

            if ret[nex] > ret[now] + 1:
                ret[nex] = ret[now] + 1
                plis[nex] = now
                q.append(nex)

    return ret,plis

N = int(stdin.readline())

lis = [ [] for i in range(N) ]

for loop in range(N-1):

    a,b = map(int,stdin.readline().split())
    a -= 1
    b -= 1

    lis[a].append(b)
    lis[b].append(a)

leaf = []
for i in range(N):
    if len(lis[i]) == 1:
        leaf.append(i)


st = leaf[0]
dis,tmp = NC_Dij(lis,st)
ld = dis[leaf[1]]

for i in range(1,len(leaf)):
    if dis[leaf[i]] != ld:
        print ("No")
        sys.exit()

st = leaf[-1]
dis,tmp = NC_Dij(lis,st)
for i in range(0,len(leaf)-1):
    if dis[leaf[i]] != ld:
        print ("No")
        sys.exit()

print ("Yes")
0