結果
問題 | No.2561 みんな大好きmod 998 |
ユーザー |
|
提出日時 | 2023-12-02 15:40:01 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 702 ms / 4,000 ms |
コード長 | 2,810 bytes |
コンパイル時間 | 15,007 ms |
コンパイル使用メモリ | 376,784 KB |
実行使用メモリ | 317,568 KB |
最終ジャッジ日時 | 2024-09-26 19:07:06 |
合計ジャッジ時間 | 24,040 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 44 |
ソースコード
// use itertools::Itertools; const MOD_998: usize = 998; const MOD_998244353: usize = 998244353; fn main() { let (n, k) = input_tuple::<usize>(); let a = input_vec::<usize>(); let mut result = 0_usize; // println!("{:?}", combination(&(0..n).collect::<Vec<_>>(), k)); // for combination in (0..n).combinations(k) { for combination in combination(&(0..n).collect::<Vec<_>>(), k) { let mut s_998 = 0_usize; let mut s_998244353 = 0_usize; for i in combination { s_998 += a[i]; s_998 %= MOD_998; s_998244353 += a[i]; s_998244353 %= MOD_998244353; } if s_998244353 <= s_998 { result += 1; result %= MOD_998; } } println!("{}", result); } fn combination(elems: &Vec<usize>, r: usize) -> Vec<Vec<usize>> { // vector<vector<T>> v{}; let mut v = Vec::new(); // vector<T> v1(r); let mut v1 = vec![0; r]; // int n = elems.size(); let n = elems.len(); rec(n, r, &elems, &mut v, &mut v1, 0, 0, 0); return v; } fn rec( n: usize, r: usize, elems: &Vec<usize>, v: &mut Vec<Vec<usize>>, v1: &mut Vec<usize>, bit: usize, now: usize, num_decided: usize, ) { if num_decided == r { // int index = 0; let mut index = 0; // for(int i=0; i<n;i++){ // if(bit&(1<<i)){ // v1[index] = elems[i]; // index++; // } // } for i in 0..n { if bit & (1 << i) > 0 { v1[index] = elems[i]; index += 1; } } // v.push_back(v1); v.push(v1.clone()); return; } // if(now == n)return; if now == n { return; } rec( n, r, &elems, v, v1, bit + (1 << now), now + 1, num_decided + 1, ); rec(n, r, &elems, v, v1, bit, now + 1, num_decided); } fn input_tuple<T>() -> (T, T) where T: std::str::FromStr, <T as std::str::FromStr>::Err: std::fmt::Debug, { let stdin = std::io::stdin(); let mut buf = String::new(); stdin.read_line(&mut buf).unwrap(); buf = buf.trim_end().to_owned(); let mut iter = buf.split_whitespace(); let n = iter.next().unwrap().parse().unwrap(); let m = iter.next().unwrap().parse().unwrap(); (n, m) } fn input_vec<T>() -> Vec<T> where T: std::str::FromStr, <T as std::str::FromStr>::Err: std::fmt::Debug, { let stdin = std::io::stdin(); let mut buf = String::new(); stdin.read_line(&mut buf).unwrap(); buf = buf.trim_end().to_owned(); let iter = buf.split_whitespace(); let line = iter.map(|x| x.parse().unwrap()).collect(); line }