結果

問題 No.2561 みんな大好きmod 998
コンテスト
ユーザー ragna
提出日時 2023-10-30 02:28:30
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
MLE  
実行時間 -
コード長 1,390 bytes
記録
コンパイル時間 1,618 ms
コンパイル使用メモリ 176,152 KB
実行使用メモリ 818,176 KB
最終ジャッジ日時 2024-09-25 22:29:32
合計ジャッジ時間 7,250 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other MLE * 1 -- * 43
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
//#include<atcoder/modint>
//using mint=atcoder::modint998244353;
using namespace std;
typedef long long ll;
#define rep(i,a,b) for(ll i=(ll)(a);i<(ll)(b);i++)
#define rrep(i,a,b) for(ll i=(ll)(a-1);i>=(ll)(b);i--)
#define MOD 998244353
//#define MOD 1000000007
#define INF 1e18
#define Pair pair<ll,ll>
//#define PI numbers::pi
//#define E numbers::e
template <typename T> bool chmax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
template <typename T> bool chmin(T& a,T b){if(a>b){a=b;return 1;}return 0;}
ll dx[8]={-2,-1,1,2,-2,-1,1,2};
ll dy[8]={-1,-2,-2,-1,1,2,2,1};

int main(){
    ll n,k; cin >> n >> k;
    vector<ll> a(n);
    rep(i,0,n) cin >> a[i];
    vector<multiset<pair<Pair,ll>>> dp(n+1);
    dp[0].insert({{0,0},0});
    rep(i,0,n){
        for(auto&& x:dp[i]){
            ll p=x.first.first,q=x.first.second,r=x.second;
            if(r==k) dp[i+1].insert(x);
            else if(n-i+r==k&&(p+a[i]%998)%998>=(q+a[i]%MOD)%MOD) dp[i+1].insert({{(p+a[i]%998)%998,(q+a[i]%MOD)%MOD},r+1});
            else{
                dp[i+1].insert(x);
                if(r<k-1||(p+a[i]%998)%998>=(q+a[i]%MOD)%MOD) dp[i+1].insert({{(p+a[i]%998)%998,(q+a[i]%MOD)%MOD},r+1});
            }
        }
    }
    ll ans=0;
    for(auto&& x:dp[n]){
        ll p=x.first.first,q=x.first.second,r=x.second;
        if(r==k&&q<=p) ans++;
    }
    cout << ans%998 << endl;
}
0