結果

問題 No.2561 みんな大好きmod 998
ユーザー LevyxLevyx
提出日時 2023-12-02 15:04:50
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 174 ms / 4,000 ms
コード長 1,590 bytes
コンパイル時間 4,851 ms
コンパイル使用メモリ 310,088 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 18:00:43
合計ジャッジ時間 8,196 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 7 ms
5,248 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 174 ms
5,376 KB
testcase_04 AC 171 ms
5,376 KB
testcase_05 AC 167 ms
5,376 KB
testcase_06 AC 170 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 29 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 170 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 1 ms
5,376 KB
testcase_19 AC 121 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 1 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 38 ms
5,376 KB
testcase_27 AC 170 ms
5,376 KB
testcase_28 AC 57 ms
5,376 KB
testcase_29 AC 14 ms
5,376 KB
testcase_30 AC 36 ms
5,376 KB
testcase_31 AC 85 ms
5,376 KB
testcase_32 AC 26 ms
5,376 KB
testcase_33 AC 14 ms
5,376 KB
testcase_34 AC 166 ms
5,376 KB
testcase_35 AC 15 ms
5,376 KB
testcase_36 AC 15 ms
5,376 KB
testcase_37 AC 7 ms
5,376 KB
testcase_38 AC 29 ms
5,376 KB
testcase_39 AC 35 ms
5,376 KB
testcase_40 AC 37 ms
5,376 KB
testcase_41 AC 29 ms
5,376 KB
testcase_42 AC 54 ms
5,376 KB
testcase_43 AC 7 ms
5,376 KB
testcase_44 AC 20 ms
5,376 KB
testcase_45 AC 123 ms
5,376 KB
testcase_46 AC 20 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, s, n) for (int i = (s); i < (int)(n); i++)
#include <atcoder/all>
using namespace atcoder;
using mint1 = modint1000000007;
using mint2 = modint998244353;
using ll = long long;
using ull = unsigned long long;
using ld = long double;
ld pi=3.141592653589793;
const int inf=2e9;
const ll linf=2e18;
const ld eps=1e-14;
#define Yes cout << "Yes" << endl
#define No cout << "No" << endl

template <typename T> bool next_combination(const T first, const T last, ll k) {
    const T subset = first + k;
    // empty container | k = 0 | k == n 
    if (first == last || first == subset || last == subset) {
        return false;
    }
    T src = subset;
    while (first != src) {
        src--;
        if (*src < *(last - 1)) {
            T dest = subset;
            while (*src >= *dest) {
                dest++;
            }
            iter_swap(src, dest);
            rotate(src + 1, dest + 1, last);
            rotate(subset, subset + (last - dest) - 1, last);
            return true;
        }
    }
    // restore
    rotate(first, subset, last);
    return false;
}

int main() {
    ll n,k;
    cin >> n >> k;
    vector<ll> a(n);
    vector<ll> p(n);
    for(int i=0; i<n; i++) {
        p[i]=i;
    }
    for(int i=0; i<n; i++) {
        cin >> a[i];
    }

    ll ans=0;
    do {
        ll s=0;
        for (int i = 0; i < k; i++) {
            s+=a[p[i]];
        }
        if(s%998244353<=s%998) {
            ans++;
        }
    } while(next_combination(p.begin(), p.end(), k));

    cout << ans%998 << endl;
}
0