結果
問題 | No.2532 Want Play More |
ユーザー | 👑 SPD_9X2 |
提出日時 | 2023-11-03 22:16:26 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,021 ms / 2,000 ms |
コード長 | 1,647 bytes |
コンパイル時間 | 297 ms |
コンパイル使用メモリ | 82,944 KB |
実行使用メモリ | 123,184 KB |
最終ジャッジ日時 | 2024-09-25 20:41:26 |
合計ジャッジ時間 | 12,013 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
""" https://yukicoder.me/problems/no/2532 """ import sys from sys import stdin from collections import deque def NC_Dij(lis,start): ret = [float("inf")] * len(lis) ret[start] = 0 visit = [] q = deque([start]) plis = [i for i in range(len(lis))] while len(q) > 0: now = q.popleft() visit.append(now) for nex in lis[now]: if ret[nex] > ret[now] + 1: ret[nex] = ret[now] + 1 plis[nex] = now q.append(nex) return ret,plis,visit N = int(stdin.readline()) lis = [ [] for i in range(N) ] for i in range(N-1): a,b = map(int,stdin.readline().split()) a -= 1 b -= 1 lis[a].append(b) lis[b].append(a) dlis,plis,visit = NC_Dij(lis,0) visit.reverse() dp = [None] * N for v in visit: turn = dlis[v] % 2 if turn == 0: #max now = 0 for nex in lis[v]: if nex != plis[v]: now = max(now , dp[nex] + 1) else: now = float("inf") for nex in lis[v]: if nex != plis[v]: now = min(now , dp[nex] + 1) if now == float("inf"): now = 0 dp[v] = now print (dp[0]) dp = [None] * N for v in visit: turn = dlis[v] % 2 if turn == 1: #max now = 0 for nex in lis[v]: if nex != plis[v]: now = max(now , dp[nex] + 1) else: now = float("inf") for nex in lis[v]: if nex != plis[v]: now = min(now , dp[nex] + 1) if now == float("inf"): now = 0 dp[v] = now print (dp[0])