結果

問題 No.2949 Product on Tree
ユーザー Carpenters-Cat
提出日時 2024-10-25 23:06:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 738 bytes
コンパイル時間 4,321 ms
コンパイル使用メモリ 255,088 KB
最終ジャッジ日時 2025-02-24 23:51:30
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 3
other RE * 46
権限があれば一括ダウンロードができます

ソースコード

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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0