結果

問題 No.2949 Product on Tree
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2024-10-25 23:06:20
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 738 bytes
コンパイル時間 4,606 ms
コンパイル使用メモリ 266,428 KB
実行使用メモリ 15,580 KB
最終ジャッジ日時 2024-10-25 23:06:52
合計ジャッジ時間 30,541 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
testcase_47 RE -
testcase_48 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
int N;
using Vi = std::vector<int> ;
using VVi = vector<Vi>;
using Vm = vector<mint>;
Vm dp1, dp2;
VVi gr;
Vm A;
void dfs(int s = 0, int p = -1) {
	dp1[s] = 0;
	dp2[s] = A[s];
	mint sum = 1;
	for (int& v : gr[s]) {
		if (v == p) continue;
		dfs(v, s);
		dp1[s] += dp1[v];
		dp2[s] += dp2[v] * A[s];
		dp1[s] += sum * dp2[v] * A[s];
		sum += dp2[v];
	}
}
int main () {
	cin >> N;
	gr.resize(N);
	A.resize(N);
	for (auto& a : A) {
		int x;
		cin >> x;
		a = x;
	}
	for (int i = 1; i < N; i ++) {
		int a, b;
		cin >> a >> b;
		gr[--a].push_back(--b);
		gr[b].push_back(a);
	}
	dfs();
	cout << dp1[0].val() << endl;
}
0