結果
| 問題 | 
                            No.1424 Ultrapalindrome
                             | 
                    
| コンテスト | |
| ユーザー | 
                             nephrologist
                         | 
                    
| 提出日時 | 2021-03-12 22:29:46 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 1,196 bytes | 
| コンパイル時間 | 299 ms | 
| コンパイル使用メモリ | 82,304 KB | 
| 実行使用メモリ | 102,112 KB | 
| 最終ジャッジ日時 | 2024-10-14 13:01:33 | 
| 合計ジャッジ時間 | 4,854 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 27 WA * 2 | 
ソースコード
import sys
from collections import deque
input = sys.stdin.buffer.readline
n = int(input())
graph = [[] for _ in range(n)]
deg1 = [0] * n
deg2 = [0] * n
for _ in range(n - 1):
    a, b = map(int, input().split())
    a, b = a - 1, b - 1
    graph[a].append(b)
    graph[b].append(a)
    deg1[a] += 1
    deg1[b] += 1
kouho = deque()
haji = []
dist1 = [-1] * n
done = [0] * n
for i in range(n):
    if deg1[i] == 1:
        haji.append(i)
def ok():
    print("Yes")
    exit()
def ng():
    print("No")
    exit()
if n == 2 or n == 3:
    ok()
# bfs
# print("kouho", kouho)
v = haji[0]
dist1[v] = 0
kouho.append(v)
while kouho:
    v = kouho.popleft()
    d = dist1[v]
    for u in graph[v]:
        if dist1[u] == -1:
            dist1[u] = d + 1
            kouho.append(u)
temp = dist1[haji[1]]
for v in haji[1:]:
    if dist1[v] != temp:
        ng()
kouho = deque()
dist2 = [-1] * n
v = haji[-1]
dist2[v] = 0
kouho.append(v)
while kouho:
    v = kouho.popleft()
    d = dist2[v]
    for u in graph[v]:
        if dist2[u] == -1:
            dist2[u] = d + 1
            kouho.append(u)
temp = dist2[haji[0]]
for v in haji[:-1]:
    if dist2[v] != temp:
        ng()
ok()
            
            
            
        
            
nephrologist