結果

問題 No.2561 みんな大好きmod 998
ユーザー arad
提出日時 2024-01-02 22:49:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 258 ms / 4,000 ms
コード長 1,396 bytes
コンパイル時間 4,170 ms
コンパイル使用メモリ 251,136 KB
最終ジャッジ日時 2025-02-18 15:57:40
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;
using VD = vector<double>;
using VVD = vector<VD>;
using VS = vector<string>;
using P = pair<ll,ll>;
using VP = vector<P>;
#define rep(i, n) for (ll i = 0; i < ll(n); i++)
#define out(x) cout << x << endl
#define dout(x) cout << fixed << setprecision(10) << x << endl
#define all(a) (a).begin(),(a).end()
#define rall(a) (a).rbegin(),(a).rend()
#define sz(x) (int)(x.size())
#define re0 return 0
#define pcnt __builtin_popcountll
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
constexpr int inf = 1e9;
constexpr ll INF = 1e18;
using mint = modint1000000007;
//using mint = modint998244353;
int di[4] = {1,0,-1,0};
int dj[4] = {0,1,0,-1};

void solve(){
    ll n,k;
    cin >> n >> k;
    VL a(n);
    rep(i,n) cin >> a[i];
    VL p;
    rep(i,n-k) p.emplace_back(0);
    rep(i,k) p.emplace_back(1);
    ll ans = 0;
    do{
        ll sum = 0;
        rep(i,n) if(p[i]) sum += a[i];
        if(sum%998244353 <= sum%998) ans++;
    }while(next_permutation(all(p)));
    ans %= 998;
    out(ans);
}

int main(){
    int t=1;
    rep(ti,t) solve();
}
0