結果

問題 No.1424 Ultrapalindrome
ユーザー AEn
提出日時 2022-11-30 23:15:16
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,144 bytes
コンパイル時間 269 ms
コンパイル使用メモリ 82,564 KB
実行使用メモリ 99,520 KB
最終ジャッジ日時 2024-10-07 18:48:49
合計ジャッジ時間 5,178 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 26 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict, deque

N = int(input())
G = [list() for _ in range(N)]

for i in range(N-1):
    a, b = map(int, input().split())
    a-=1;b-=1
    G[a].append(b)
    G[b].append(a)

s = []
for i in range(N):
    if len(G[i])==1:
        s.append(i)
if len(s)==2:
    print('Yes')
else:
    dis = [-1]*N
    pre = [-1]*N
    d = deque()
    d.append(s[0])
    dis[s[0]] = 0
    while d:
        v = d.popleft()
        for nex in G[v]:
            if dis[nex]==-1:
                dis[nex]=dis[v]+1
                pre[nex] = v
                d.append(nex)
    if dis[s[1]]%2==1:
        exit(print('No'))
    else:
        dd = dis[s[1]]//2
        nn = dis[s[1]]
        v = s[1]
        while nn>dd:
            v = pre[v]
            nn = dis[v]
        d.append(v)
        dis2 = [-1]*N
        dis2[v] = 0
        while d:
            ver = d.popleft()
            for nex in G[ver]:
                if dis2[nex]==-1:
                    dis2[nex]=dis2[ver]+1
                    d.append(nex)
        for ss in s:
            if dis2[ss]!=nn:
                print('No')
                exit()
        print('Yes')
0