結果

問題 No.2949 Product on Tree
ユーザー pengin_2000pengin_2000
提出日時 2024-10-25 22:35:14
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 487 ms / 2,000 ms
コード長 2,386 bytes
コンパイル時間 385 ms
コンパイル使用メモリ 31,616 KB
実行使用メモリ 20,648 KB
最終ジャッジ日時 2024-10-25 22:35:49
合計ジャッジ時間 21,180 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
14,192 KB
testcase_01 AC 3 ms
14,196 KB
testcase_02 AC 3 ms
14,192 KB
testcase_03 AC 452 ms
20,536 KB
testcase_04 AC 391 ms
20,368 KB
testcase_05 AC 410 ms
20,544 KB
testcase_06 AC 429 ms
20,556 KB
testcase_07 AC 403 ms
20,380 KB
testcase_08 AC 417 ms
20,464 KB
testcase_09 AC 434 ms
20,492 KB
testcase_10 AC 434 ms
20,440 KB
testcase_11 AC 428 ms
20,388 KB
testcase_12 AC 426 ms
20,492 KB
testcase_13 AC 385 ms
20,456 KB
testcase_14 AC 363 ms
20,224 KB
testcase_15 AC 392 ms
20,324 KB
testcase_16 AC 426 ms
20,352 KB
testcase_17 AC 432 ms
20,392 KB
testcase_18 AC 389 ms
20,396 KB
testcase_19 AC 448 ms
20,280 KB
testcase_20 AC 432 ms
20,552 KB
testcase_21 AC 460 ms
20,464 KB
testcase_22 AC 390 ms
20,444 KB
testcase_23 AC 415 ms
20,432 KB
testcase_24 AC 487 ms
20,408 KB
testcase_25 AC 434 ms
20,636 KB
testcase_26 AC 411 ms
20,632 KB
testcase_27 AC 412 ms
20,628 KB
testcase_28 AC 403 ms
20,576 KB
testcase_29 AC 408 ms
20,588 KB
testcase_30 AC 457 ms
20,532 KB
testcase_31 AC 457 ms
20,504 KB
testcase_32 AC 425 ms
20,636 KB
testcase_33 AC 451 ms
20,504 KB
testcase_34 AC 417 ms
20,540 KB
testcase_35 AC 442 ms
20,580 KB
testcase_36 AC 452 ms
20,628 KB
testcase_37 AC 445 ms
20,648 KB
testcase_38 AC 401 ms
20,536 KB
testcase_39 AC 458 ms
20,564 KB
testcase_40 AC 425 ms
20,504 KB
testcase_41 AC 413 ms
20,540 KB
testcase_42 AC 475 ms
20,628 KB
testcase_43 AC 137 ms
19,688 KB
testcase_44 AC 140 ms
19,504 KB
testcase_45 AC 179 ms
20,504 KB
testcase_46 AC 161 ms
20,136 KB
testcase_47 AC 115 ms
19,324 KB
testcase_48 AC 171 ms
20,196 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<stdio.h>
long long int u[400005], v[400005];
long long int h[400005], l, z;
long long int depth[200005], par[200005];
long long int comp_h(long long int a, long long int b)
{
	if (z == 0)
	{
		if (u[h[a]] > u[h[b]])
			return 1;
		else
			return -1;
	}
	else
	{
		if (depth[h[a]] < depth[h[b]])
			return 1;
		else
			return -1;
	}
}
void swap_h(long long int a, long long int b)
{
	long long int f = h[a];
	h[a] = h[b];
	h[b] = f;
	return;
}
void push(long long int ne)
{
	h[l] = ne;
	long long int p = l;
	l++;
	for (; p > 0; p = (p - 1) / 2)
		if (comp_h((p - 1) / 2, p) > 0)
			swap_h((p - 1) / 2, p);
	return;
}
long long int pop()
{
	l--;
	swap_h(0, l);
	long long int p = 0;
	for (;;)
	{
		if (2 * p + 2 < l)
		{
			if (comp_h(2 * p + 1, 2 * p + 2) > 0)
			{
				if (comp_h(p, 2 * p + 2) > 0)
					swap_h(p, 2 * p + 2);
				p = 2 * p + 2;
			}
			else
			{
				if (comp_h(p, 2 * p + 1) > 0)
					swap_h(p, 2 * p + 1);
				p = 2 * p + 1;
			}
		}
		else if (2 * p + 1 < l)
		{
			if (comp_h(p, 2 * p + 1) > 0)
				swap_h(p, 2 * p + 1);
			p = 2 * p + 1;
		}
		else
			break;
	}
	return h[l];
}
long long int a[200005];
long long int c[400005];
long long int dp[200005];
int main()
{
	long long int n;
	scanf("%lld", &n);
	long long int i;
	for (i = 0; i < n; i++)
		scanf("%lld", &a[i]);
	long long int m = n - 1;
	for (i = 0; i < m; i++)
	{
		scanf("%lld %lld", &u[i], &v[i]);
		u[i]--;
		v[i]--;
		u[i + m] = v[i];
		v[i + m] = u[i];
	}
	const long long int p = 998244353;
	m *= 2;
	l = 0;
	z = 0;
	for (i = 0; i < m; i++)
		push(i);
	for (i = 0; i < m; i++)
		c[i] = pop();
	c[m] = m;
	u[m] = -1;
	for (i = 0; i < n; i++)
		depth[i] = -1;
	int min, mid, max;
	depth[0] = par[0] = 0;
	h[0] = 0;
	l = 1;
	while (l > 0)
	{
		l--;
		i = h[l];
		min = -1;
		max = m;
		while (max - min > 1)
		{
			mid = (max + min) / 2;
			if (u[c[mid]] < i)
				min = mid;
			else
				max = mid;
		}
		for (; u[c[max]] == i; max++)
		{
			if (depth[v[c[max]]] < 0)
			{
				depth[v[c[max]]] = depth[i] + 1;
				par[v[c[max]]] = i;
				h[l] = v[c[max]];
				l++;
			}
		}
	}
	for (i = 0; i < n; i++)
		dp[i] = 1;
	z = 1;
	l = 0;
	for (i = 0; i < n; i++)
		push(i);
	long long int ans = 0;
	while (l > 1)
	{
		i = pop();
		dp[i] = dp[i] * a[i] % p;
		ans += dp[par[i]] * dp[i] % p * a[par[i]] % p;
		ans %= p;
		dp[par[i]] += dp[i];
		dp[par[i]] %= p;
	}
	printf("%ld\n", ans);
	return 0;
}
0