結果
問題 | No.2598 Kadomatsu on Tree |
ユーザー | rlangevin |
提出日時 | 2024-01-05 12:34:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 579 ms / 2,000 ms |
コード長 | 1,248 bytes |
コンパイル時間 | 537 ms |
コンパイル使用メモリ | 81,572 KB |
実行使用メモリ | 137,604 KB |
最終ジャッジ日時 | 2024-01-05 12:35:06 |
合計ジャッジ時間 | 22,689 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 37 ms
53,460 KB |
testcase_01 | AC | 37 ms
53,460 KB |
testcase_02 | AC | 36 ms
53,460 KB |
testcase_03 | AC | 37 ms
53,460 KB |
testcase_04 | AC | 37 ms
53,460 KB |
testcase_05 | AC | 38 ms
53,460 KB |
testcase_06 | AC | 45 ms
53,460 KB |
testcase_07 | AC | 37 ms
53,460 KB |
testcase_08 | AC | 39 ms
53,460 KB |
testcase_09 | AC | 48 ms
53,460 KB |
testcase_10 | AC | 37 ms
53,460 KB |
testcase_11 | AC | 37 ms
53,460 KB |
testcase_12 | AC | 40 ms
53,460 KB |
testcase_13 | AC | 93 ms
76,708 KB |
testcase_14 | AC | 103 ms
76,844 KB |
testcase_15 | AC | 91 ms
76,728 KB |
testcase_16 | AC | 97 ms
76,840 KB |
testcase_17 | AC | 89 ms
76,480 KB |
testcase_18 | AC | 101 ms
76,844 KB |
testcase_19 | AC | 96 ms
76,812 KB |
testcase_20 | AC | 90 ms
76,504 KB |
testcase_21 | AC | 97 ms
76,588 KB |
testcase_22 | AC | 107 ms
76,568 KB |
testcase_23 | AC | 499 ms
131,216 KB |
testcase_24 | AC | 476 ms
131,756 KB |
testcase_25 | AC | 515 ms
131,760 KB |
testcase_26 | AC | 469 ms
131,348 KB |
testcase_27 | AC | 491 ms
131,480 KB |
testcase_28 | AC | 498 ms
131,628 KB |
testcase_29 | AC | 522 ms
131,892 KB |
testcase_30 | AC | 483 ms
131,216 KB |
testcase_31 | AC | 497 ms
131,760 KB |
testcase_32 | AC | 538 ms
131,492 KB |
testcase_33 | AC | 508 ms
132,032 KB |
testcase_34 | AC | 520 ms
131,076 KB |
testcase_35 | AC | 523 ms
132,300 KB |
testcase_36 | AC | 541 ms
132,164 KB |
testcase_37 | AC | 498 ms
132,032 KB |
testcase_38 | AC | 504 ms
132,172 KB |
testcase_39 | AC | 545 ms
132,172 KB |
testcase_40 | AC | 502 ms
132,172 KB |
testcase_41 | AC | 507 ms
132,172 KB |
testcase_42 | AC | 504 ms
132,172 KB |
testcase_43 | AC | 482 ms
132,172 KB |
testcase_44 | AC | 510 ms
132,172 KB |
testcase_45 | AC | 483 ms
132,172 KB |
testcase_46 | AC | 533 ms
132,428 KB |
testcase_47 | AC | 517 ms
132,172 KB |
testcase_48 | AC | 548 ms
132,844 KB |
testcase_49 | AC | 112 ms
79,940 KB |
testcase_50 | AC | 340 ms
109,692 KB |
testcase_51 | AC | 148 ms
84,816 KB |
testcase_52 | AC | 204 ms
94,988 KB |
testcase_53 | AC | 178 ms
105,504 KB |
testcase_54 | AC | 171 ms
99,300 KB |
testcase_55 | AC | 100 ms
82,564 KB |
testcase_56 | AC | 236 ms
120,576 KB |
testcase_57 | AC | 135 ms
92,068 KB |
testcase_58 | AC | 520 ms
131,484 KB |
testcase_59 | AC | 456 ms
131,508 KB |
testcase_60 | AC | 550 ms
137,604 KB |
testcase_61 | AC | 579 ms
132,028 KB |
testcase_62 | AC | 557 ms
134,936 KB |
ソースコード
import sys input = sys.stdin.readline N = int(input()) G = [[] for i in range(N)] for i in range(N - 1): u, v = map(int, input().split()) u, v = u - 1, v - 1 G[u].append(v) G[v].append(u) A = list(map(int, input().split())) mod = 998244353 def non_rec_dfs(s): stack = [] stack.append(s) par = [-1] * N sz = [1] * N ans = 0 while stack: u = stack.pop() if u >= 0: stack.append(~u) for v in G[u]: if v == par[u]: continue par[v] = u stack.append(v) else: u = ~u for v in G[u]: if v == par[u]: continue sz[u] += sz[v] p2, p1, m2, m1 = 0, 0, 0, 0 for v in G[u]: val = sz[v] if v == par[u]: val = N - sz[u] if A[v] > A[u]: p2 += val ** 2 p1 += val elif A[v] < A[u]: m2 += val ** 2 m1 += val ans += (p1**2 - p2)//2 + (m1**2 - m2)//2 ans %= mod return ans print(non_rec_dfs(0))