結果

問題 No.2561 みんな大好きmod 998
ユーザー miiitomi
提出日時 2023-12-02 14:52:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 612 bytes
コンパイル時間 2,017 ms
コンパイル使用メモリ 197,148 KB
最終ジャッジ日時 2025-02-18 04:01:12
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll MOD = 998244352;

int main() {
    ll N, K;
    cin >> N >> K;
    vector<ll> A(N);
    for (int i = 0; i < N; i++) cin >> A[i];

    vector<ll> v(N, 0);
    for (int k = 0; k < K; k++) v[N-1-k] = 1;

    ll cnt = 0;
    do {
        ll s1 = 0, s2 = 0;
        for (int i = 0; i < N; i++) {
            if (v[i] == 0) continue;
            s1 = (s1 + A[i]) % 998;
            s2 = (s2 + A[i]) % MOD;
        }
        if (s2 <= s1) cnt = (cnt + 1) % 998;
    } while (next_permutation(v.begin(), v.end()));

    cout << cnt << endl;
}
0