結果

問題 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;
}
0