結果

問題 No.2561 みんな大好きmod 998
ユーザー cumene_rxcumene_rx
提出日時 2023-12-02 15:01:59
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 266 ms / 4,000 ms
コード長 3,985 bytes
コンパイル時間 5,252 ms
コンパイル使用メモリ 317,304 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 17:55:57
合計ジャッジ時間 9,342 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 12 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 266 ms
5,376 KB
testcase_04 AC 249 ms
5,376 KB
testcase_05 AC 250 ms
5,376 KB
testcase_06 AC 252 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 39 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 250 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 181 ms
5,376 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 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 56 ms
5,376 KB
testcase_27 AC 248 ms
5,376 KB
testcase_28 AC 73 ms
5,376 KB
testcase_29 AC 21 ms
5,376 KB
testcase_30 AC 60 ms
5,376 KB
testcase_31 AC 132 ms
5,376 KB
testcase_32 AC 33 ms
5,376 KB
testcase_33 AC 22 ms
5,376 KB
testcase_34 AC 254 ms
5,376 KB
testcase_35 AC 19 ms
5,376 KB
testcase_36 AC 19 ms
5,376 KB
testcase_37 AC 10 ms
5,376 KB
testcase_38 AC 41 ms
5,376 KB
testcase_39 AC 62 ms
5,376 KB
testcase_40 AC 63 ms
5,376 KB
testcase_41 AC 41 ms
5,376 KB
testcase_42 AC 77 ms
5,376 KB
testcase_43 AC 10 ms
5,376 KB
testcase_44 AC 29 ms
5,376 KB
testcase_45 AC 179 ms
5,376 KB
testcase_46 AC 24 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
#include <random>
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef unsigned long long ull;
using mint9 = modint998244353;
using mint1 = modint1000000007;
//using mint = modint;
const ll INF = 1000000007;
const ll LINF = INF * INF;
//const long long LLMAX = 9223372036854775807;
const long long LLMAX = 100000000000000000;

const double TIME_1 = 1.5;
//const ll MOD = 998244353;
std::mt19937 get_rand_mt;
#define PA pair<int,int>
#define PAL pair<ll,ll>
#define TP tuple<int,int,int>
#define TPL tuple<ll, ll, ll>
#define rep(i, z, n) for(int i = z; i < n; i++)
#define repl(i, z, n) for(ll i = z; i < n; i++)
#define revrep(i, z, n) for(int i = n - 1; i >= z; i--)
#define revrepl(i, z, n) for(ll i = n - 1; i >= z; i--)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define SZ(x) ((int)(x).size())
#define PB push_back
#define YN(T) if(T){cout<<"Yes"<<endl;}else{cout<<"No"<<endl;}
#define YN2(T) if(T){cout<<"YES"<<endl;}else{cout<<"NO"<<endl;}
#define OK cout << "OK" << endl;
#define FX(x) cout << fixed << setprecision(10) << x << endl;
#define shuffle(x) shuffle(all(x), get_rand_mt)
#define PQ priority_queue<int,vector<int>,greater<int>>
#define PQL priority_queue<ll,vector<ll>,greater<ll>>
#define PQPA priority_queue<PA,vector<PA>,greater<PA>>

template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
template<class T>T ceilX(T a, T b) {return (a + b - 1)/b; }
ll lmax(ll a, ll b){if(a>b) return a; return b;}
ll lmin(ll a, ll b){if(a>b) return b; return a;}
ll comb(ll X, ll Y){ll res = 1LL;ll R = X;repl(i, 1LL, lmin(Y + 1LL, X - Y + 1LL)){res *= (R - i + 1LL);res /= i;}return res;}
template<class T>void Rekkyo(vector<T> X){cout<<"Rekkyo"<<endl;for(auto Y:X){cout<<Y<<" ";}cout<<endl<<"end"<<endl;}
template<class T>void Rekkyo1(vector<T> X){cout<<"Rekkyo1"<<endl;int cnt=0;for(auto Y:X){cout<<cnt<<" "<<Y<<endl;cnt++;}cout<<"end"<<endl;}
template<class T>void Rekkyo2(vector<vector<T>> X){cout<<"Rekkyo2"<<endl;for(auto Y:X){for(auto Z:Y){cout <<Z<<" ";}cout<<endl;}cout<<"end"<<endl;}

vector<int> Eratosthenes(const int N){
    vector<bool> is_prime(N + 1);
    rep(i, 0, N + 1) is_prime[i] = true;
    vector<int> P;
    rep(i, 2, N + 1){
        if(is_prime[i]){
            for( int j = 2 * i; j <= N; j += i ) is_prime[ j ] = false;
            P.emplace_back(i);
        }
    }
    return P;
}
vector<vector<int>> make_edge(vector<int> u, vector<int> v, bool angle){
    int N = u.size();
    vector<vector<int>> res;
    rep(i, 0, N){res[u[i]].PB(v[i]);if(!angle) res[v[i]].PB(u[i]);}
    return res;
}
vector<vector<PA>> make_edge(vector<int> u, vector<int> v, vector<int> w, bool angle){
    int N = u.size();
    vector<vector<PA>> res;
    rep(i, 0, N){res[u[i]].PB(PA(v[i], w[i]));if(!angle) res[v[i]].PB(PA(u[i], w[i]));}
    return res;
}
vector<vector<PAL>> make_edgeL(vector<int> u, vector<int> v, vector<ll> w, bool angle){
    int N = u.size();
    vector<vector<PAL>> res;
    rep(i, 0, N){res[u[i]].PB(PAL(v[i], w[i]));if(!angle) res[v[i]].PB(PAL(u[i], w[i]));}
    return res;
}









int main(){
    int N, K;
    cin >> N >> K;
    vector<ll> A1(N), A2(N);
    rep(i, 0, N){
        int a;
        cin >> a;
        A1[i] = a;
        A2[i] = a;
        A1[i] %= 998;
        A2[i] %= 998244353;
    }
    ll ans = 0;
    
    vector<int> p;
    rep(i, 0, N - K){
        p.PB(0);
    }
    rep(i, 0, K){
        p.PB(1);
    }
    
    do{
        //Rekkyo(p);
        ll B1 = 0, B2 = 0;
        rep(i, 0, N){
            if(p[i] == 1){
                B1 += A1[i];
                B1 %= 998;
                B2 += A2[i];
                B2 %= 998244353;
            }
        }
        if(B2 <= B1){
            ans++;
            ans %= 998;
        }
    }while(next_permutation(all(p)));

    cout << ans << endl;
}
0