結果

問題 No.2561 みんな大好きmod 998
ユーザー ななな
提出日時 2023-12-02 20:09:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,140 bytes
コンパイル時間 1,945 ms
コンパイル使用メモリ 170,576 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 21:36:52
合計ジャッジ時間 5,692 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 42 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define int long long
#define rep(i, n) for (int i = 0; i < n; i++)
template <typename T>
bool next_combination(const T first, const T last, int k) {
  const T subset = first + k;
  // empty container | k = 0 | k == n
  if (first == last || first == subset || last == subset) {
    return false;
  }
  T src = subset;
  while (first != src) {
    src--;
    if (*src < *(last - 1)) {
      T dest = subset;
      while (*src >= *dest) {
        dest++;
      }
      iter_swap(src, dest);
      rotate(src + 1, dest + 1, last);
      rotate(subset, subset + (last - dest) - 1, last);
      return true;
    }
  }
  // restore
  rotate(first, subset, last);
  return false;
}

signed main() {
  int n, k;
  cin >> n >> k;
  vector<int> a(n);
  vector<int> c(n);
  rep(i, n) cin >> a[i];
  rep(i, n) c[i] = i;
  int ans = 0;
  do {
    int p = 0;
    int p1 = 0;
    rep(i, k) {
      p += a[c[i]];
      p1 += a[c[i]];
      p %= 998;
      p %= 998244353;
    }

    if (p1 <= p)
      ans++;
    ans %= 998;

  } while (next_combination(c.begin(), c.end(), k));
  cout << ans % 998 << endl;
}
0