結果

問題 No.2561 みんな大好きmod 998
ユーザー ragnaragna
提出日時 2023-10-30 02:28:30
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 133 ms
34,560 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 MLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

#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