結果

問題 No.2561 みんな大好きmod 998
ユーザー takumi152
提出日時 2023-12-02 14:50:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 44 ms / 4,000 ms
コード長 973 bytes
コンパイル時間 928 ms
コンパイル使用メモリ 90,320 KB
最終ジャッジ日時 2025-02-18 03:58:56
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
#include <utility>
#include <string>
#include <queue>
#include <stack>

using namespace std;

typedef long long int ll;
typedef pair<int, int> Pii;

const ll mod_small = 998;
const ll mod_large = 998244353;

ll solve(vector<int> &idx, int next_begin, int sum_small, int sum_large, const int n, const int k, const vector<ll> &a) {
  if ((int) idx.size() == k) {
    if (sum_large <= sum_small) return 1;
    else return 0;
  }
  ll ans = 0;
  for (int i = next_begin; i < n; i++) {
    idx.push_back(i);
    ans += solve(idx, i + 1, (sum_small + a[i]) % mod_small, (sum_large + a[i]) % mod_large, n, k, a);
    idx.pop_back();
  }
  return ans;
}

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);

  int n, k;
  cin >> n >> k;
  vector<ll> a(n);
  for (auto &x: a) cin >> x;

  vector<int> idx;
  ll ans = solve(idx, 0, 0, 0, n, k, a);

  cout << ans % mod_small << endl;

  return 0;
}
0