結果

問題 No.2949 Product on Tree
ユーザー NakLon131NakLon131
提出日時 2024-10-26 10:51:24
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 161 ms / 2,000 ms
コード長 1,733 bytes
コンパイル時間 12,475 ms
コンパイル使用メモリ 401,052 KB
実行使用メモリ 35,068 KB
最終ジャッジ日時 2024-10-26 10:51:52
合計ジャッジ時間 22,006 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,820 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 0 ms
6,816 KB
testcase_03 AC 133 ms
24,712 KB
testcase_04 AC 124 ms
24,240 KB
testcase_05 AC 120 ms
24,844 KB
testcase_06 AC 121 ms
24,872 KB
testcase_07 AC 134 ms
24,396 KB
testcase_08 AC 152 ms
24,924 KB
testcase_09 AC 147 ms
25,128 KB
testcase_10 AC 142 ms
25,432 KB
testcase_11 AC 139 ms
26,000 KB
testcase_12 AC 147 ms
27,440 KB
testcase_13 AC 145 ms
27,632 KB
testcase_14 AC 141 ms
28,600 KB
testcase_15 AC 144 ms
30,304 KB
testcase_16 AC 152 ms
29,056 KB
testcase_17 AC 150 ms
29,660 KB
testcase_18 AC 150 ms
29,472 KB
testcase_19 AC 135 ms
30,932 KB
testcase_20 AC 134 ms
30,808 KB
testcase_21 AC 139 ms
31,296 KB
testcase_22 AC 142 ms
31,412 KB
testcase_23 AC 131 ms
25,224 KB
testcase_24 AC 142 ms
25,236 KB
testcase_25 AC 145 ms
25,188 KB
testcase_26 AC 137 ms
25,144 KB
testcase_27 AC 134 ms
25,224 KB
testcase_28 AC 139 ms
25,268 KB
testcase_29 AC 139 ms
25,440 KB
testcase_30 AC 138 ms
25,852 KB
testcase_31 AC 137 ms
26,772 KB
testcase_32 AC 151 ms
26,976 KB
testcase_33 AC 148 ms
29,380 KB
testcase_34 AC 144 ms
32,504 KB
testcase_35 AC 143 ms
30,396 KB
testcase_36 AC 145 ms
33,552 KB
testcase_37 AC 154 ms
33,876 KB
testcase_38 AC 141 ms
31,744 KB
testcase_39 AC 161 ms
32,000 KB
testcase_40 AC 142 ms
35,012 KB
testcase_41 AC 137 ms
33,108 KB
testcase_42 AC 139 ms
35,068 KB
testcase_43 AC 53 ms
19,704 KB
testcase_44 AC 55 ms
19,944 KB
testcase_45 AC 69 ms
25,220 KB
testcase_46 AC 63 ms
22,708 KB
testcase_47 AC 46 ms
17,152 KB
testcase_48 AC 63 ms
23,424 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn main() {
    input!{
		n: usize,
		a: [usize; n],
		uv: [(Usize1, Usize1); n-1],
	}

	let mut g = vec![Vec::new(); n];
	for &(u, v) in &uv {
		g[u].push(v);
		g[v].push(u);
	}

	let mut ans = 0;
	dfs(0, MOD93, &g, &a, &mut ans);
	println!("{}", ans);

	fn dfs(pos: usize, ppos: usize, g: &Vec<Vec<usize>>, a: &Vec<usize>, ans: &mut usize) -> usize {
		let mut sm1 = 0;
		let mut sm2 = 0;
		for &npos in &g[pos] {
			if npos == ppos { continue; }
			let esm = dfs(npos, pos, g, a, ans) % MOD93;
			sm1 += esm; 		sm1 %= MOD93;
			sm2 += esm * esm; 	sm2 %= MOD93;
		}

		// ans+=a[p]*(sm1+(sm1*sm1-sm2)/2);
		let multi = (sm1*sm1+MOD93-sm2) % MOD93;
		let divs = div(multi, 2, MOD93);
		*ans += a[pos]* (sm1+divs); *ans %= MOD93;

		a[pos]*(sm1+1) % MOD93
	}
}

// 繰り返し二乗法
fn mod_pow(mut a: usize, mut b: usize, mods: usize) -> usize {
	let mut res = 1;
	while b > 0 {
		if b % 2 == 1 {
			res *= a;
			res %= mods;
		}
		a *= a;
		a %= mods;
		b >>= 1;
	}
	res
}

// 繰り返し二乗法による除算
fn div(a: usize, b: usize, mods: usize) -> usize {
	let mut ans = a;
	let x = mod_pow(b, mods - 2, mods);
	ans *= x;
	ans %= mods;
	ans
}

// const MOD17: usize = 1000000007;
const MOD93: usize = 998244353;
// const INF: usize = 1 << 60;
// let dx = vec![!0, 0, 1, 0]; // 上左下右
// let dy = vec![0, !0, 0, 1]; // 上左下右
// let d = vec!{(!0, 0), (0, !0), (1, 0), (0, 1)}; // 上左下右

#[allow(unused)]
use proconio::{input, marker::Chars, marker::Usize1};

#[allow(unused)]
use std::{
	mem::swap,
	cmp::min, cmp::max,
	cmp::Reverse,
	collections::HashSet, collections::BTreeSet,
	collections::HashMap, collections::BTreeMap,
	collections::BinaryHeap,
	collections::VecDeque,
	iter::FromIterator,
};
0