結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #

// 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
}
0