結果

問題 No.1424 Ultrapalindrome
ユーザー kuro_Bkuro_B
提出日時 2023-04-06 14:50:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 709 ms / 2,000 ms
コード長 2,675 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 82,152 KB
実行使用メモリ 276,848 KB
最終ジャッジ日時 2024-10-02 13:48:51
合計ジャッジ時間 9,246 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 125 ms
82,952 KB
testcase_01 AC 112 ms
82,992 KB
testcase_02 AC 109 ms
83,464 KB
testcase_03 AC 112 ms
83,276 KB
testcase_04 AC 110 ms
83,340 KB
testcase_05 AC 111 ms
83,212 KB
testcase_06 AC 112 ms
83,196 KB
testcase_07 AC 111 ms
83,156 KB
testcase_08 AC 109 ms
82,792 KB
testcase_09 AC 296 ms
108,048 KB
testcase_10 AC 309 ms
107,124 KB
testcase_11 AC 252 ms
102,244 KB
testcase_12 AC 304 ms
106,416 KB
testcase_13 AC 188 ms
94,368 KB
testcase_14 AC 289 ms
104,168 KB
testcase_15 AC 123 ms
88,920 KB
testcase_16 AC 241 ms
102,076 KB
testcase_17 AC 265 ms
104,124 KB
testcase_18 AC 234 ms
99,096 KB
testcase_19 AC 346 ms
104,304 KB
testcase_20 AC 209 ms
93,316 KB
testcase_21 AC 360 ms
112,948 KB
testcase_22 AC 253 ms
97,564 KB
testcase_23 AC 145 ms
90,532 KB
testcase_24 AC 225 ms
97,080 KB
testcase_25 AC 203 ms
93,456 KB
testcase_26 AC 361 ms
106,208 KB
testcase_27 AC 340 ms
112,428 KB
testcase_28 AC 471 ms
240,780 KB
testcase_29 AC 709 ms
276,848 KB
testcase_30 AC 249 ms
111,880 KB
testcase_31 AC 235 ms
111,996 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

###スニペット始まり###
import sys, re
from copy import copy, deepcopy
from math import ceil, floor, sqrt,factorial, gcd, pi, degrees, radians, sin, asin, cos, acos, tan, atan2
from collections import Counter, deque, defaultdict
from heapq import heapify, heappop, heappush
from itertools import permutations, accumulate, product, combinations, combinations_with_replacement
from bisect import bisect, bisect_left, bisect_right
from functools import reduce, lru_cache
from string import ascii_uppercase, ascii_lowercase
from decimal import Decimal, ROUND_HALF_UP #四捨五入用

def input(): return sys.stdin.readline().rstrip('\n')
#easy-testのpypyでは再帰数を下げる。
if __file__=='prog.py':
    sys.setrecursionlimit(10**5)
else:
    sys.setrecursionlimit(10**6)

def lcm(a, b): return a * b // gcd(a, b)

#3つ以上の最大公約数/最小公倍数
def gcd_v2(l: list): return reduce(gcd, l)
def lcm_v2(l: list): return reduce(lcm, l)

#nPk
def nPk(n, k): return factorial(n) // factorial(n - k)

#逆元
def modinv(a, mod=10**9+7): return pow(a, mod-2, mod)
INF = float('inf')
MOD = 10 ** 9 + 7
###スニペット終わり###
N=int(input())
graph = [[] for _ in range(N)]
for _ in range(N-1):
    a, b = map(int, input().split())
    graph[a-1].append(b-1)
    graph[b-1].append(a-1)  # 有向グラフなら消す

dists=[-1]*N
leaf_cnt=0
def dfs(s, dist):
    global leaf_cnt
    dists[s]=dist
    flag=True
    for v in graph[s]:
        if dists[v]==-1:
            flag=False
            dfs(v, dist+1)
    if flag:
        leaf_cnt+=1
dfs(0,0)

s=dists.index(max(dists)) #0から一番遠い点を求める。
dists=[-1]*N
prev=defaultdict(int)
def dfs2(s, dist):
    dists[s]=dist
    for v in graph[s]:
        if dists[v]==-1:
            dfs2(v, dist+1)
            prev[v]=s
dfs2(s,0)
diam=max(dists) #直径
diam_leaf=dists.index(diam) #直径の末端。

if diam%2!=0:
    
    if leaf_cnt==1:
        #もし、葉がひとつならYes
        print('Yes')
    else:
        print('No')
else:
    #直径が偶数なら可能性あり。
    #まずは中心を求める。
    center=diam_leaf
    for _ in range(diam//2):
        center=prev[center]
    dist_set=set()
    visited=[False]*N
    def dfs3(s, dist):
        visited[s]=True
        flag=True
        cnt=0
        for v in graph[s]:
            if not visited[v]:
                cnt+=1
                flag=False
                dfs3(v, dist+1)
        if s!=center and cnt>=2:
            print('No')
            exit()
        if flag:
            dist_set.add(dist)
    dfs3(center,0)
    if len(dist_set)==1:
        print('Yes')
    else:
        print('No')
0