結果

問題 No.2561 みんな大好きmod 998
ユーザー strawberry0929
提出日時 2023-12-02 14:59:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 114 ms / 4,000 ms
コード長 1,761 bytes
コンパイル時間 3,748 ms
コンパイル使用メモリ 230,676 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 17:49:13
合計ジャッジ時間 6,137 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using namespace std;
using mint = modint998244353;
using ll = long long;
using lb = long double;
using VI = vector<int>;
using VL = vector<ll>;
using P = pair<int, int>;
inline void scan(){}
template<class Head,class... Tail>
inline void scan(Head&head,Tail&... tail){std::cin>>head;scan(tail...);}
#define LL(...) ll __VA_ARGS__;scan(__VA_ARGS__)
#define LB(...) lb __VA_ARGS__;scan(__VA_ARGS__)
#define INT(...) int __VA_ARGS__;scan(__VA_ARGS__)
#define STR(...) string __VA_ARGS__;scan(__VA_ARGS__)
template<class ll> inline bool chmax(ll& a, ll b) { if (a < b) { a = b; return 1; } return 0; }template<class ll> inline bool chmin(ll& a, ll b) { if (a > b) { a = b; return 1; } return 0; }
template<typename T>
std::istream &operator>>(std::istream&is,std::vector<T>&v){for(T &in:v){is>>in;}return is;}
template<typename T>
std::ostream &operator<<(std::ostream&os,const std::vector<T>&v){for(auto it=std::begin(v);it!=std::end(v);){os<<*it<<((++it)!=std::end(v)?" ":"");}return os;}
#define rep(i, s, n) for (int i = (s); i < (int)(n); i++)
#define ALL(a)  (a).begin(),(a).end()
#define RALL(a)  (a).rbegin(),(a).rend()
const int inf = INT_MAX / 2; const ll infl = 1LL << 60;
//_
int n, k;
vector<ll> a;
vector<int> x;
int ans = 0;
void dfs(int v){
  x.push_back(v);
  if(x.size() == k){
    ll s1 = 0;
    mint s2 = 0;
    for(auto&& i : x){
      s1 += a[i];
      s2 += a[i];
      s1 %= 998;
    }
    if(s2.val() <= s1){
      ans++;
    }
  }else{
    rep(i, v + 1, n){
      dfs(i);
      x.pop_back();
    }
  }
}
int main(){
  cin >> n >> k;
  a.resize(n);
  cin >> a;
  rep(i, 0, n){
    dfs(i);
    //cout << x << endl;
    x.pop_back();
  }
  cout << ans % 998 << endl;
}
0