結果

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

ソースコード

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 t = 0;
    rep(i, k) { t += a[c[i]]; }

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

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