結果
| 問題 | 
                            No.1424 Ultrapalindrome
                             | 
                    
| コンテスト | |
| ユーザー | 
                             brthyyjp
                         | 
                    
| 提出日時 | 2021-03-12 23:33:04 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 129 ms / 2,000 ms | 
| コード長 | 777 bytes | 
| コンパイル時間 | 215 ms | 
| コンパイル使用メモリ | 82,176 KB | 
| 実行使用メモリ | 103,256 KB | 
| 最終ジャッジ日時 | 2024-10-14 16:27:10 | 
| 合計ジャッジ時間 | 3,917 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 29 | 
ソースコード
import sys
import io, os
input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline
n = int(input())
g = [[] for i in range(n)]
for i in range(n-1):
    a, b = map(int, input().split())
    a, b = a-1, b-1
    g[a].append(b)
    g[b].append(a)
G = []
L = []
for i in range(n):
    if len(g[i]) >= 3:
        G.append(i)
    if len(g[i]) == 1:
        L.append(i)
if len(G) >= 2:
    print('No')
    exit()
if not G:
    print('Yes')
    exit()
s = G[0]
from collections import deque
q = deque([])
q.append(s)
dist = [-1]*n
dist[s] = 0
while q:
    v = q.popleft()
    for u in g[v]:
        if dist[u] == -1:
            dist[u] = dist[v]+1
            q.append(u)
D = set()
for l in L:
    D.add(dist[l])
#print(D)
if len(D) == 1:
    print('Yes')
else:
    print('No')
            
            
            
        
            
brthyyjp