結果
問題 | No.2561 みんな大好きmod 998 |
ユーザー |
|
提出日時 | 2023-12-02 14:57:55 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 171 ms / 4,000 ms |
コード長 | 1,210 bytes |
コンパイル時間 | 1,492 ms |
コンパイル使用メモリ | 170,252 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-26 17:46:16 |
合計ジャッジ時間 | 4,618 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 44 |
ソースコード
#include <bits/stdc++.h> using namespace std; 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; } int main() { long ans = 0; int N, K; cin >> N >> K; vector<int> A(N), p(N); for (int i = 0; i < N; i++) { p[i] = i; } for (int i = 0; i < N; i++) { cin >> A[i]; } do { long S = 0; for (int i = 0; i < K; i++) { S += A[p[i]]; } if (S % 998244353 <= S % 998) { ans++; } } while (next_combination(p.begin(), p.end(), K)); cout << ans % 998 << endl; }