結果
問題 | No.763 Noelちゃんと木遊び |
ユーザー | roaris |
提出日時 | 2019-08-18 19:25:32 |
言語 | Python3 (3.11.6 + numpy 1.26.0 + scipy 1.11.3) |
結果 |
AC
|
実行時間 | 476 ms / 2,000 ms |
コード長 | 869 bytes |
コンパイル時間 | 894 ms |
コンパイル使用メモリ | 10,880 KB |
実行使用メモリ | 64,652 KB |
最終ジャッジ日時 | 2023-07-24 18:55:42 |
合計ジャッジ時間 | 8,991 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 476 ms
64,652 KB |
testcase_01 | AC | 148 ms
18,956 KB |
testcase_02 | AC | 353 ms
32,416 KB |
testcase_03 | AC | 247 ms
25,696 KB |
testcase_04 | AC | 167 ms
20,340 KB |
testcase_05 | AC | 232 ms
25,312 KB |
testcase_06 | AC | 447 ms
39,476 KB |
testcase_07 | AC | 437 ms
38,660 KB |
testcase_08 | AC | 255 ms
26,468 KB |
testcase_09 | AC | 164 ms
19,764 KB |
testcase_10 | AC | 73 ms
13,156 KB |
testcase_11 | AC | 463 ms
39,884 KB |
testcase_12 | AC | 401 ms
34,420 KB |
testcase_13 | AC | 391 ms
34,080 KB |
testcase_14 | AC | 349 ms
31,668 KB |
testcase_15 | AC | 234 ms
25,572 KB |
testcase_16 | AC | 53 ms
11,460 KB |
testcase_17 | AC | 241 ms
25,480 KB |
testcase_18 | AC | 463 ms
39,876 KB |
testcase_19 | AC | 409 ms
34,620 KB |
testcase_20 | AC | 411 ms
34,904 KB |
ソースコード
from collections import defaultdict import sys input = sys.stdin.readline sys.setrecursionlimit(10 ** 6) def dfs(cur, prev, state): if memo[cur][state] != -1: return memo[cur][state] if state == 0: res = 0 for child in adj_dict[cur]: if prev != child: res += max(dfs(child, cur, 0), dfs(child, cur, 1)) else: res = 1 for child in adj_dict[cur]: if prev != child: res += max(dfs(child, cur, 0), dfs(child, cur, 1) - 1) memo[cur][state] = res return res N = int(input()) adj_dict = defaultdict(list) for _ in range(N-1): Ui, Vi = map(int, input().split()) adj_dict[Ui-1].append(Vi-1) adj_dict[Vi-1].append(Ui-1) memo = [[-1, -1] for _ in range(N)] ans = max(dfs(0, None, 0), dfs(0, None, 1)) print(ans)