#934217 (C++23) No.2561 みんな大好きmod 998

提出ソース

結果

問題	No.2561 みんな大好きmod 998
ユーザー	cumene_rx
提出日時	2023-12-02 15:01:59
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	271 ms / 4,000 ms
コード長	3,985 bytes
コンパイル時間	5,544 ms
コンパイル使用メモリ	318,108 KB
実行使用メモリ	6,676 KB
最終ジャッジ日時	2023-12-02 15:02:16
合計ジャッジ時間	9,902 ms
ジャッジサーバーID （参考情報）	judge9 / judge10

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,676 KB
testcase_01	AC	12 ms 6,676 KB
testcase_02	AC	2 ms 6,676 KB
testcase_03	AC	271 ms 6,676 KB
testcase_04	AC	260 ms 6,676 KB
testcase_05	AC	261 ms 6,676 KB
testcase_06	AC	260 ms 6,676 KB
testcase_07	AC	2 ms 6,676 KB
testcase_08	AC	2 ms 6,676 KB
testcase_09	AC	3 ms 6,676 KB
testcase_10	AC	2 ms 6,676 KB
testcase_11	AC	41 ms 6,676 KB
testcase_12	AC	2 ms 6,676 KB
testcase_13	AC	2 ms 6,676 KB
testcase_14	AC	2 ms 6,676 KB
testcase_15	AC	261 ms 6,676 KB
testcase_16	AC	2 ms 6,676 KB
testcase_17	AC	2 ms 6,676 KB
testcase_18	AC	2 ms 6,676 KB
testcase_19	AC	189 ms 6,676 KB
testcase_20	AC	2 ms 6,676 KB
testcase_21	AC	2 ms 6,676 KB
testcase_22	AC	2 ms 6,676 KB
testcase_23	AC	2 ms 6,676 KB
testcase_24	AC	2 ms 6,676 KB
testcase_25	AC	2 ms 6,676 KB
testcase_26	AC	57 ms 6,676 KB
testcase_27	AC	261 ms 6,676 KB
testcase_28	AC	76 ms 6,676 KB
testcase_29	AC	22 ms 6,676 KB
testcase_30	AC	64 ms 6,676 KB
testcase_31	AC	134 ms 6,676 KB
testcase_32	AC	33 ms 6,676 KB
testcase_33	AC	22 ms 6,676 KB
testcase_34	AC	261 ms 6,676 KB
testcase_35	AC	20 ms 6,676 KB
testcase_36	AC	20 ms 6,676 KB
testcase_37	AC	11 ms 6,676 KB
testcase_38	AC	42 ms 6,676 KB
testcase_39	AC	64 ms 6,676 KB
testcase_40	AC	64 ms 6,676 KB
testcase_41	AC	42 ms 6,676 KB
testcase_42	AC	75 ms 6,676 KB
testcase_43	AC	10 ms 6,676 KB
testcase_44	AC	30 ms 6,676 KB
testcase_45	AC	189 ms 6,676 KB
testcase_46	AC	26 ms 6,676 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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;
}

yukicoder

結果

テストケース

ソースコード