結果
問題 |
No.1424 Ultrapalindrome
|
ユーザー |
![]() |
提出日時 | 2021-03-13 05:12:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 148 ms / 2,000 ms |
コード長 | 1,321 bytes |
コンパイル時間 | 202 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 99,116 KB |
最終ジャッジ日時 | 2024-10-14 17:53:10 |
合計ジャッジ時間 | 4,366 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
import sys from collections import deque sys.setrecursionlimit(10**7) def I(): return int(sys.stdin.readline().rstrip()) def MI(): return map(int,sys.stdin.readline().rstrip().split()) def LI(): return list(map(int,sys.stdin.readline().rstrip().split())) def LI2(): return list(map(int,sys.stdin.readline().rstrip())) def S(): return sys.stdin.readline().rstrip() def LS(): return list(sys.stdin.readline().rstrip().split()) def LS2(): return list(sys.stdin.readline().rstrip()) N = I() Graph = [[] for _ in range(N+1)] deg = [0]*(N+1) # deg[i] = 頂点iの次数 for _ in range(N-1): a,b = MI() Graph[a].append(b) Graph[b].append(a) deg[a] += 1 deg[b] += 1 count = [0]*N # count[i] = 次数iの頂点数 for i in range(1,N+1): count[deg[i]] += 1 c = sum(count[3:]) if c == 0: print('Yes') exit() if c >= 2: print('No') exit() start = 0 for i in range(1,N+1): if deg[i] >= 3: start = i break dist = [-1]*(N+1) dist[start] = 0 deq = deque([start]) while deq: i = deq.pop() for j in Graph[i]: if dist[j] == -1: dist[j] = dist[i]+1 deq.appendleft(j) leaf = [] for i in range(1,N+1): if deg[i] == 1: leaf.append(i) X = set([dist[i] for i in leaf]) if len(X) == 1: print('Yes') else: print('No')