結果

問題 No.2598 Kadomatsu on Tree
ユーザー hiro1729hiro1729
提出日時 2024-01-01 09:47:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,867 ms / 2,000 ms
コード長 790 bytes
コンパイル時間 351 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 321,204 KB
最終ジャッジ日時 2024-09-29 11:33:51
合計ジャッジ時間 43,798 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,404 KB
testcase_01 AC 38 ms
52,764 KB
testcase_02 AC 39 ms
52,336 KB
testcase_03 AC 38 ms
54,204 KB
testcase_04 AC 38 ms
52,940 KB
testcase_05 AC 39 ms
52,824 KB
testcase_06 AC 39 ms
52,488 KB
testcase_07 AC 42 ms
52,820 KB
testcase_08 AC 38 ms
53,228 KB
testcase_09 AC 38 ms
52,064 KB
testcase_10 AC 38 ms
53,080 KB
testcase_11 AC 38 ms
52,560 KB
testcase_12 AC 38 ms
53,596 KB
testcase_13 AC 165 ms
78,504 KB
testcase_14 AC 155 ms
79,240 KB
testcase_15 AC 167 ms
78,952 KB
testcase_16 AC 185 ms
79,280 KB
testcase_17 AC 141 ms
77,716 KB
testcase_18 AC 162 ms
79,152 KB
testcase_19 AC 165 ms
78,912 KB
testcase_20 AC 145 ms
78,164 KB
testcase_21 AC 146 ms
78,488 KB
testcase_22 AC 162 ms
79,028 KB
testcase_23 AC 846 ms
131,656 KB
testcase_24 AC 864 ms
124,984 KB
testcase_25 AC 848 ms
124,972 KB
testcase_26 AC 863 ms
132,116 KB
testcase_27 AC 856 ms
131,856 KB
testcase_28 AC 788 ms
125,352 KB
testcase_29 AC 827 ms
125,304 KB
testcase_30 AC 966 ms
132,084 KB
testcase_31 AC 917 ms
124,996 KB
testcase_32 AC 923 ms
132,388 KB
testcase_33 AC 881 ms
129,196 KB
testcase_34 AC 1,053 ms
131,684 KB
testcase_35 AC 871 ms
129,928 KB
testcase_36 AC 880 ms
128,276 KB
testcase_37 AC 810 ms
128,508 KB
testcase_38 AC 923 ms
130,436 KB
testcase_39 AC 879 ms
130,836 KB
testcase_40 AC 891 ms
130,076 KB
testcase_41 AC 1,053 ms
130,720 KB
testcase_42 AC 865 ms
129,952 KB
testcase_43 AC 909 ms
130,336 KB
testcase_44 AC 917 ms
131,636 KB
testcase_45 AC 950 ms
130,468 KB
testcase_46 AC 975 ms
130,288 KB
testcase_47 AC 882 ms
129,932 KB
testcase_48 AC 1,867 ms
305,044 KB
testcase_49 AC 277 ms
95,872 KB
testcase_50 AC 1,202 ms
253,500 KB
testcase_51 AC 468 ms
124,928 KB
testcase_52 AC 666 ms
150,656 KB
testcase_53 AC 305 ms
101,808 KB
testcase_54 AC 260 ms
96,196 KB
testcase_55 AC 174 ms
83,076 KB
testcase_56 AC 418 ms
117,388 KB
testcase_57 AC 217 ms
89,824 KB
testcase_58 AC 773 ms
131,912 KB
testcase_59 AC 627 ms
125,172 KB
testcase_60 AC 1,795 ms
297,036 KB
testcase_61 AC 1,862 ms
304,068 KB
testcase_62 AC 1,787 ms
321,204 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys; sys.setrecursionlimit(2 * 10 ** 5)

N = int(input())
g = [[] for _ in range(N)]
for _ in range(N - 1):
	u, v = map(int, input().split())
	u -= 1
	v -= 1
	g[u].append(v)
	g[v].append(u)
a = list(map(int, input().split()))


vis = [False] * N
ans = 0
def dfs(u):
	global ans
	d = []
	vis[u] = True
	res = 0
	s = 1
	j = -1
	l = []
	r = []
	for i in g[u]:
		if vis[i]:
			if a[i] > a[u]:
				j = len(l)
				l.append(0)
			if a[i] < a[u]:
				j = len(r) + N
				r.append(0)
		else:
			t = dfs(i)
			s += t
			if a[i] > a[u]:
				l.append(t)
			if a[i] < a[u]:
				r.append(t)
	if j != -1:
		if j < N:
			l[j] = N - s
		else:
			r[j - N] = N - s
	ans += (sum(l) ** 2 - sum(i ** 2 for i in l)) // 2 + (sum(r) ** 2 - sum(i ** 2 for i in r)) // 2
	return s

dfs(0)
print(ans % 998244353)
0