結果

問題 No.2677 Minmax Independent Set
コンテスト
ユーザー nu50218
提出日時 2023-11-30 23:39:22
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,247 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 82,152 KB
実行使用メモリ 589,528 KB
最終ジャッジ日時 2024-09-30 00:04:49
合計ジャッジ時間 16,973 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 9 TLE * 3 MLE * 1 -- * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/pypy3

import sys
sys.setrecursionlimit(10**6)

N = int(input())

to = [[] for i in range(N)]

for i in range(N-1):
    x, y = map(int, input().split())
    x -= 1
    y -= 1
    to[x].append(y)
    to[y].append(x)

# https://snuke.hatenablog.com/entry/2019/01/16/203626 を改変

pi = [0] * N
deg = [0] * N
dp = [[] for i in range(N)]
ans = [0] * N

def dfs(v, p=-1):
  deg[v] = len(to[v])
  res = (0, 0)
  dp[v] = [0]*deg[v]
  for i in range(deg[v]):
    u = to[v][i]
    if u == p:
      pi[v] = i
      continue
    dp[v][i] = dfs(u, v)
    val = (max(dp[v][i]), dp[v][i][0])
    res = (res[0] + val[0], res[1] + val[1])
  return (res[0], res[1] + 1)

def bfs(v, res_p=0, p=-1):
  if p != -1:
    dp[v][pi[v]] = res_p
  dpl = [(0, 0)]*(deg[v]+1)
  for i in range(deg[v]):
    dpl[i+1] = (dpl[i][0] + max(dp[v][i]), dpl[i][1] + dp[v][i][0])
  dpr = [(0, 0)]*(deg[v]+1)
  for i in range(deg[v]-1,-1,-1):
    dpr[i] = (dpr[i+1][0] + max(dp[v][i]), dpr[i+1][1] + dp[v][i][0])
  ans[v] = (dpr[0][0], dpr[0][1] + 1)
  for i in range(deg[v]):
    u = to[v][i]
    if u == p:
      continue
    val = (dpl[i][0] + dpr[i+1][0], dpl[i][1] + dpr[i+1][1] + 1)
    bfs(u, val, v)

dfs(0)
bfs(0)

# 答え出力

print(min([x[1] for x in ans]))
0