結果

問題 No.2598 Kadomatsu on Tree
ユーザー hiro1729
提出日時 2024-01-01 09:47:39
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 778 bytes
コンパイル時間 437 ms
コンパイル使用メモリ 82,328 KB
実行使用メモリ 321,436 KB
最終ジャッジ日時 2024-09-27 17:18:37
合計ジャッジ時間 35,273 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 WA * 36
権限があれば一括ダウンロードができます

ソースコード

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)
0