結果
問題 | No.1810 RGB Biscuits |
ユーザー | phspls |
提出日時 | 2022-10-27 11:14:22 |
言語 | Rust (1.77.0) |
結果 |
AC
|
実行時間 | 24 ms / 2,000 ms |
コード長 | 1,949 bytes |
コンパイル時間 | 1,025 ms |
コンパイル使用メモリ | 145,712 KB |
実行使用メモリ | 4,380 KB |
最終ジャッジ日時 | 2023-09-18 04:54:45 |
合計ジャッジ時間 | 2,690 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 1 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,380 KB |
testcase_05 | AC | 21 ms
4,380 KB |
testcase_06 | AC | 24 ms
4,376 KB |
testcase_07 | AC | 23 ms
4,380 KB |
testcase_08 | AC | 24 ms
4,380 KB |
testcase_09 | AC | 23 ms
4,376 KB |
testcase_10 | AC | 15 ms
4,380 KB |
testcase_11 | AC | 3 ms
4,380 KB |
testcase_12 | AC | 2 ms
4,380 KB |
testcase_13 | AC | 1 ms
4,376 KB |
testcase_14 | AC | 4 ms
4,376 KB |
testcase_15 | AC | 1 ms
4,376 KB |
testcase_16 | AC | 2 ms
4,380 KB |
testcase_17 | AC | 12 ms
4,376 KB |
testcase_18 | AC | 12 ms
4,376 KB |
testcase_19 | AC | 1 ms
4,376 KB |
testcase_20 | AC | 1 ms
4,380 KB |
testcase_21 | AC | 3 ms
4,376 KB |
ソースコード
const MOD: usize = 1e9 as usize + 7; fn power_mat(target: &Vec<Vec<usize>>, times: usize) -> Vec<Vec<usize>> { let n = target.len(); if times == 0 { let mut ret = vec![vec![0usize; n]; n]; for i in 0..n { ret[i][i] = 1; } return ret; } if times == 1 { return target.clone(); } let temp = power_mat(target, times/2); if times % 2 == 1 { calc_mat(&calc_mat(&temp, &temp), target) } else { calc_mat(&temp, &temp) } } fn calc_mat(left: &Vec<Vec<usize>>, right: &Vec<Vec<usize>>) -> Vec<Vec<usize>> { let sizei = left.len(); let sizej = right[0].len(); let sizek = left[0].len(); let mut ret = vec![vec![0usize; sizej]; sizei]; for i in 0..sizei { for j in 0..sizej { ret[i][j] = (0..sizek).map(|k| left[i][k] * right[k][j] % MOD).sum::<usize>() % MOD; } } ret } fn main() { let mut ab = String::new(); std::io::stdin().read_line(&mut ab).ok(); let ab: Vec<usize> = ab.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let a = ab[0]; let b = ab[1]; let mut n = String::new(); std::io::stdin().read_line(&mut n).ok(); let n: usize = n.trim().parse().unwrap(); let queries = (0..n).map(|_| { let mut temp = String::new(); std::io::stdin().read_line(&mut temp).ok(); let temp: usize = temp.trim().parse().unwrap(); temp }) .collect::<Vec<_>>(); let op1 = vec![vec![1,0,0], vec![0,1,0], vec![a,b,1]]; let op2 = vec![vec![0,0,1],vec![1,0,0],vec![0,0,0]]; let op = calc_mat(&op2, &op1); let base = vec![vec![1],vec![1],vec![0]]; for &t in queries.iter() { let op = power_mat(&op, t/2); let op = if t % 2 == 0 { op } else { calc_mat(&op1, &op) }; let res = calc_mat(&op, &base); println!("{}", res.iter().map(|v| v[0]).sum::<usize>() % MOD); } }