結果
問題 | No.2561 みんな大好きmod 998 |
ユーザー |
|
提出日時 | 2023-12-02 15:04:50 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 174 ms / 4,000 ms |
コード長 | 1,590 bytes |
コンパイル時間 | 4,851 ms |
コンパイル使用メモリ | 310,088 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-26 18:00:43 |
合計ジャッジ時間 | 8,196 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 44 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define rep(i, s, n) for (int i = (s); i < (int)(n); i++) #include <atcoder/all> using namespace atcoder; using mint1 = modint1000000007; using mint2 = modint998244353; using ll = long long; using ull = unsigned long long; using ld = long double; ld pi=3.141592653589793; const int inf=2e9; const ll linf=2e18; const ld eps=1e-14; #define Yes cout << "Yes" << endl #define No cout << "No" << endl template <typename T> bool next_combination(const T first, const T last, ll 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() { ll n,k; cin >> n >> k; vector<ll> a(n); vector<ll> p(n); for(int i=0; i<n; i++) { p[i]=i; } for(int i=0; i<n; i++) { cin >> a[i]; } ll ans=0; do { ll 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; }