結果

問題 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