結果
問題 | No.2561 みんな大好きmod 998 |
ユーザー |
![]() |
提出日時 | 2023-12-02 15:17:48 |
言語 | Rust (1.77.0 + proconio) |
結果 |
MLE
|
実行時間 | - |
コード長 | 6,353 bytes |
コンパイル時間 | 18,554 ms |
コンパイル使用メモリ | 377,364 KB |
実行使用メモリ | 803,284 KB |
最終ジャッジ日時 | 2024-09-26 18:25:22 |
合計ジャッジ時間 | 20,794 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 31 ms
26,624 KB |
testcase_02 | AC | 1 ms
5,376 KB |
testcase_03 | MLE | - |
testcase_04 | MLE | - |
testcase_05 | MLE | - |
testcase_06 | MLE | - |
testcase_07 | AC | 2 ms
6,944 KB |
testcase_08 | AC | 2 ms
6,940 KB |
testcase_09 | AC | 3 ms
6,944 KB |
testcase_10 | AC | 2 ms
6,940 KB |
testcase_11 | AC | 147 ms
110,820 KB |
testcase_12 | AC | 2 ms
6,944 KB |
testcase_13 | AC | 1 ms
6,944 KB |
testcase_14 | AC | 2 ms
6,944 KB |
testcase_15 | MLE | - |
testcase_16 | AC | 2 ms
6,944 KB |
testcase_17 | AC | 15 ms
6,944 KB |
testcase_18 | AC | 1 ms
6,940 KB |
testcase_19 | MLE | - |
testcase_20 | AC | 2 ms
6,940 KB |
testcase_21 | AC | 2 ms
6,944 KB |
testcase_22 | AC | 2 ms
6,940 KB |
testcase_23 | AC | 2 ms
6,944 KB |
testcase_24 | AC | 1 ms
6,940 KB |
testcase_25 | AC | 2 ms
6,944 KB |
testcase_26 | AC | 189 ms
161,456 KB |
testcase_27 | MLE | - |
testcase_28 | AC | 287 ms
217,708 KB |
testcase_29 | AC | 67 ms
53,632 KB |
testcase_30 | AC | 199 ms
169,828 KB |
testcase_31 | AC | 445 ms
380,680 KB |
testcase_32 | AC | 115 ms
99,928 KB |
testcase_33 | AC | 67 ms
53,964 KB |
testcase_34 | MLE | - |
testcase_35 | AC | 68 ms
57,756 KB |
testcase_36 | AC | 67 ms
58,184 KB |
testcase_37 | AC | 164 ms
23,168 KB |
testcase_38 | AC | 125 ms
111,200 KB |
testcase_39 | AC | 191 ms
170,196 KB |
testcase_40 | AC | 190 ms
169,804 KB |
testcase_41 | AC | 128 ms
110,780 KB |
testcase_42 | AC | 250 ms
217,832 KB |
testcase_43 | AC | 27 ms
23,168 KB |
testcase_44 | AC | 90 ms
82,220 KB |
testcase_45 | MLE | - |
testcase_46 | AC | 84 ms
75,528 KB |
ソースコード
use std::io::stdin; struct Problem { n: usize, k: usize, xs: Vec<i64>, } impl Problem { fn read<R: IProconReader>(mut r: R) -> Problem { let (n, k) = r.read_usize_2(); let xs = r.read_vec_any(); Problem { n, k, xs } } fn solve(&self) -> Answer { let comb = combinations(self.n, self.k); let ans = comb.iter().filter(|bs|{ let sum = self.xs.iter().zip(bs.iter()).map(|(x, b)| x * b).sum::<i64>(); let sum_998244353 = sum % 998244353; let sum_998 = sum % 998; sum_998244353 <= sum_998 }).count() as i64; let ans = ans % 998; Answer { ans } } } fn combinations(n: usize, k: usize) -> Vec<Vec<i64>> { let mut ret = Vec::new(); // 0 を n-k 個, 1をk個並べる let mut seq01 = std::iter::repeat(0).take(n - k).chain(std::iter::repeat(1).take(k)).collect::<Vec<_>>(); while{ ret.push(seq01.clone()); next_permutation(&mut seq01) }{}; ret } // https://ngtkana.hatenablog.com/entry/2021/11/08/000209 pub fn next_permutation<T: Ord>(a: &mut [T]) -> bool { if a.is_empty() { return false; } let Some(i) = (0..a.len() - 1).rfind(|&i| a[i] < a[i + 1]) else { return false }; let j = a.iter().rposition(|x| x > &a[i]).unwrap(); a.swap(i, j); a[i + 1..].reverse(); true } #[derive(Clone, Debug, PartialEq, Eq)] struct Answer { ans: i64, } impl Answer { fn print(&self) { println!("{}", self.ans); } } fn main() { Problem::read(ProconReader::new(stdin().lock())).solve().print(); } #[cfg(test)] mod tests { use super::*; #[allow(dead_code)] fn check(input: &str, expected: Answer) { let actual = Problem::read(ProconReader::new(input.as_bytes())).solve(); assert_eq!(expected, actual); } #[test] fn test_problem() { let _input = " 3 4 " .trim(); // check(_input, Answer { ans: 7 }); } } // ====== snippet ====== #[allow(unused_imports)] use myio::*; pub mod myio { use std::io::BufRead; pub trait IProconReader { fn read_line(&mut self) -> String; fn read_bytes(&mut self) -> Vec<u8> { self.read_line().as_bytes().to_vec() } fn read_any_1<T>(&mut self) -> T where T: std::str::FromStr, T::Err: std::fmt::Debug, { let buf = self.read_line(); buf.parse::<T>().unwrap() } fn read_any_2<T0, T1>(&mut self) -> (T0, T1) where T0: std::str::FromStr, T0::Err: std::fmt::Debug, T1: std::str::FromStr, T1::Err: std::fmt::Debug, { let buf = self.read_line(); let splitted = buf.trim().split(' ').collect::<Vec<_>>(); let a0 = splitted[0].parse::<T0>().unwrap(); let a1 = splitted[1].parse::<T1>().unwrap(); (a0, a1) } fn read_any_3<T0, T1, T2>(&mut self) -> (T0, T1, T2) where T0: std::str::FromStr, T0::Err: std::fmt::Debug, T1: std::str::FromStr, T1::Err: std::fmt::Debug, T2: std::str::FromStr, T2::Err: std::fmt::Debug, { let buf = self.read_line(); let splitted = buf.trim().split(' ').collect::<Vec<_>>(); let a0 = splitted[0].parse::<T0>().unwrap(); let a1 = splitted[1].parse::<T1>().unwrap(); let a2 = splitted[2].parse::<T2>().unwrap(); (a0, a1, a2) } fn read_any_4<T0, T1, T2, T3>(&mut self) -> (T0, T1, T2, T3) where T0: std::str::FromStr, T0::Err: std::fmt::Debug, T1: std::str::FromStr, T1::Err: std::fmt::Debug, T2: std::str::FromStr, T2::Err: std::fmt::Debug, T3: std::str::FromStr, T3::Err: std::fmt::Debug, { let buf = self.read_line(); let splitted = buf.trim().split(' ').collect::<Vec<_>>(); let a0 = splitted[0].parse::<T0>().unwrap(); let a1 = splitted[1].parse::<T1>().unwrap(); let a2 = splitted[2].parse::<T2>().unwrap(); let a3 = splitted[3].parse::<T3>().unwrap(); (a0, a1, a2, a3) } fn read_vec_any<T>(&mut self) -> Vec<T> where T: std::str::FromStr, T::Err: std::fmt::Debug, { let buf = self.read_line(); buf.trim().split(' ').map(|s| s.parse::<T>().unwrap()).collect::<Vec<T>>() } fn read_vec_i64(&mut self) -> Vec<i64> { self.read_vec_any::<i64>() } fn read_vec_usize(&mut self) -> Vec<usize> { self.read_vec_any::<usize>() } fn read_vec_str(&mut self) -> Vec<String> { self.read_vec_any::<String>() } fn read_i64_1(&mut self) -> i64 { self.read_any_1::<i64>() } fn read_i64_2(&mut self) -> (i64, i64) { self.read_any_2::<i64, i64>() } fn read_i64_3(&mut self) -> (i64, i64, i64) { self.read_any_3::<i64, i64, i64>() } fn read_i64_4(&mut self) -> (i64, i64, i64, i64) { self.read_any_4::<i64, i64, i64, i64>() } fn read_usize_1(&mut self) -> usize { self.read_any_1::<usize>() } fn read_usize_2(&mut self) -> (usize, usize) { self.read_any_2::<usize, usize>() } fn read_usize_3(&mut self) -> (usize, usize, usize) { self.read_any_3::<usize, usize, usize>() } fn read_usize_4(&mut self) -> (usize, usize, usize, usize) { self.read_any_4::<usize, usize, usize, usize>() } } pub struct ProconReader<R: BufRead> { buf_read: R, } impl<R: BufRead> ProconReader<R> { pub fn new(buf_read: R) -> ProconReader<R> { ProconReader { buf_read } } } impl<R: BufRead> IProconReader for ProconReader<R> { fn read_line(&mut self) -> String { let mut buffer = String::new(); self.buf_read.read_line(&mut buffer).unwrap(); buffer.trim().to_string() } } }