結果

問題 No.1424 Ultrapalindrome
ユーザー tobusakanatobusakana
提出日時 2022-10-14 21:34:38
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,087 bytes
コンパイル時間 218 ms
コンパイル使用メモリ 82,480 KB
実行使用メモリ 110,080 KB
最終ジャッジ日時 2024-06-26 13:17:11
合計ジャッジ時間 5,338 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

N = int(input())
G = [[] for i in range(N)]
indegree = [0] * N
for _ in range(N - 1):
    a,b = map(int,input().split())
    G[a - 1].append(b - 1)
    G[b - 1].append(a - 1)
    indegree[a - 1] += 1
    indegree[b - 1] += 1
    
# 次数1のある点から、他の全ての点への距離が同じであればOK
# を二回やって成功すればOK
start = -1
start1 = -1
for v in range(N):
    if indegree[v] == 1 and start != -1:
        start1 = v
        break
    if indegree[v] == 1:
        start = v

def check(x):
    stack = []
    dist = [-1] * N
    stack.append([start, 0])
    while stack:
        v, d = stack.pop()
        if dist[v] != -1:
            continue
        dist[v] = d
        for child in G[v]:
            if dist[child] != -1:
                continue
            stack.append([child, d + 1])
        
    ds = set()
    for v in range(N):
        if v == start:
            continue
        if indegree[v] == 1:
            ds.add(dist[v])
            
    return len(ds) == 1

if check(start) and check(start1):
    print("Yes")
else:
    print("No")
0