結果

問題 No.2561 みんな大好きmod 998
コンテスト
ユーザー srjywrdnprkt
提出日時 2023-12-07 14:08:13
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 607 ms / 4,000 ms
コード長 1,060 bytes
記録
コンパイル時間 2,245 ms
コンパイル使用メモリ 199,228 KB
最終ジャッジ日時 2025-02-18 08:57:06
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

//mi~mxからN個を重複なしで選ぶ
vector<vector<int>> combinations(int N, int mi, int mx){
    vector<vector<int>> res;
    vector<int> idx;

    auto dfs=[&](auto self, int N, int mi, int mx, vector<vector<int>> &A)->void{
        if (idx.size() == N){
            A.push_back(idx);
            return;
        }
        else{
            for (int i = mi; i <= mx; i++){
                idx.push_back(i);
                self(self, N, i+1, mx, A);
                idx.pop_back();
            }
        }
    };
    dfs(dfs, N, mi, mx, res);
    return res;
}


int main(){
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);
    int N, K, ans=0;
    cin >> N >> K;
    ll S;
    vector<ll> A(N);
    for (int i=0; i<N; i++) cin >> A[i];
    vector<vector<int>> v = combinations(K, 0, N-1);
    for (auto &x : v){
        S = 0;
        for (auto y : x) S += A[y];
        if (S % 998244353 <= S % 998) ans++;
    }
    ans %= 998;
    cout << ans << endl;

    return 0;
}
0