結果
問題 | No.2561 みんな大好きmod 998 |
ユーザー |
![]() |
提出日時 | 2023-11-18 15:46:20 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 30 ms / 4,000 ms |
コード長 | 1,628 bytes |
コンパイル時間 | 15,155 ms |
コンパイル使用メモリ | 400,080 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-26 05:49:43 |
合計ジャッジ時間 | 15,469 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 44 |
ソースコード
macro_rules! input {($($r:tt)*) => {let stdin = std::io::stdin();let mut bytes = std::io::Read::bytes(std::io::BufReader::new(stdin.lock()));let mut next = move || -> String{bytes.by_ref().map(|r|r.unwrap() as char).skip_while(|c|c.is_whitespace()).take_while(|c|!c.is_whitespace()).collect()};input_inner!{next, $($r)*}};}macro_rules! input_inner {($next:expr) => {};($next:expr,) => {};($next:expr, $var:ident : $t:tt $($r:tt)*) => {let $var = read_value!($next, $t);input_inner!{$next $($r)*}};}macro_rules! read_value {($next:expr, ( $($t:tt),* )) => { ($(read_value!($next, $t)),*) };($next:expr, [ $t:tt ; $len:expr ]) => {(0..$len).map(|_| read_value!($next, $t)).collect::<Vec<_>>()};($next:expr, $t:ty) => ($next().parse::<$t>().expect("Parse error"));}struct Data {n: usize,k: usize,a: Vec<usize>,}fn dfs(data: &Data, sum: usize, cnt: usize, idx: usize, ans: &mut usize) {if cnt == data.k {let sm = sum % 998;let lg = sum % 998244353;if lg <= sm {*ans += 1;*ans %= 998;}return;}if idx == data.n {return;}for i in idx..data.n {dfs(data, sum + data.a[i], cnt + 1, i + 1, ans);}}fn main() {input! {n: usize,k: usize,a: [usize; n],}let data = Data { n, k, a };let mut ans = 0;dfs(&data, 0, 0, 0, &mut ans);println!("{}", ans);}