結果

問題 No.2360 Path to Integer
ユーザー Carpenters-CatCarpenters-Cat
提出日時 2023-06-24 19:32:48
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 232 ms / 2,500 ms
コード長 1,564 bytes
コンパイル時間 2,405 ms
コンパイル使用メモリ 214,396 KB
実行使用メモリ 15,896 KB
最終ジャッジ日時 2024-07-01 22:32:13
合計ジャッジ時間 4,932 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 4 ms
6,940 KB
testcase_08 AC 20 ms
6,944 KB
testcase_09 AC 232 ms
15,132 KB
testcase_10 AC 198 ms
15,892 KB
testcase_11 AC 203 ms
15,896 KB
testcase_12 AC 174 ms
15,176 KB
testcase_13 AC 230 ms
15,260 KB
testcase_14 AC 217 ms
15,088 KB
testcase_15 AC 197 ms
15,128 KB
testcase_16 AC 184 ms
15,080 KB
権限があれば一括ダウンロードができます

ソースコード

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