結果

問題 No.2360 Path to Integer
ユーザー Carpenters-Cat
提出日時 2023-06-24 19:32:48
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 210 ms / 2,500 ms
コード長 1,564 bytes
コンパイル時間 2,183 ms
コンパイル使用メモリ 206,328 KB
最終ジャッジ日時 2025-02-15 02:08:21
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using Vl = vector<ll>;
using Vi = vector<int>;
using VVi = vector<Vi>;
ll m = 998244353;
int main () {
    int N;
    cin >> N;
    std::vector<string> S(N);
    for (auto& a : S) cin >> a;
    Vi tpr, par(N, -1);
    VVi gr(N);
    for (int i = 1; i < N; i ++) {
        int a, b;
        cin >> a >> b;
        gr[--a].push_back(--b);
        gr[b].push_back(a);
    }
    queue<int> que;
    que.push(0);
    while (!que.empty()) {
        int u = que.front();
        que.pop();
        tpr.push_back(u);
        for (auto v : gr[u]) {
            if (v == par[u]) continue;
            par[v] = u;
            que.push(v);
        }
    }
    for (int i = 0; i < N / 2; i ++) swap(tpr[i], tpr[N - i - 1]);
    // return 0;
    Vl dp(N), dp2(N), nm(N);
    // return 0;
    for (auto i : tpr) {
        // cout << i << endl;
        dp[i] = 0;
        nm[i] = 1;
        for (auto v : gr[i]) {
            if (v == par[i]) continue;
            dp[i] = (dp[i] + dp[v]) % m;
            nm[i] += nm[v];
        }
        for (int j = 0; j < S[i].size(); j ++) dp[i] = (dp[i] * 10) % m;
        ll ps = stoll(S[i]) % m;
        ps = (ps * nm[i]) % m;
        dp[i] = (dp[i] + ps) % m;
    }
    // return 0;
    dp2[0] = dp[0];
    for (int i = N - 2; i >= 0; i --) {
        int u = tpr[i];
        int p = par[u];
        ll x = dp[u];
        for (int j = 0; j < S[p].size(); j ++) x = (x * 10) % m;
        x = (x + nm[u] * stoll(S[p])) % m;
        dp2[u] = (dp2[p] + m - x) % m;
        for (int j = 0; j < S[u].size(); j ++) dp2[u] = (dp2[u] * 10) % m;
        dp2[u] = (dp2[u] + (N - nm[u]) * stoll(S[u])) % m;
        dp2[u] = (dp2[u] + dp[u]) % m;
    }
    // return 0;
    ll ans = 0;
    for (auto a : dp2) ans = (ans + a) % m;
    cout << ans << endl;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0