結果
問題 | No.1554 array_and_me |
ユーザー |
![]() |
提出日時 | 2021-06-19 15:47:39 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 47 ms / 2,000 ms |
コード長 | 1,736 bytes |
コンパイル時間 | 3,787 ms |
コンパイル使用メモリ | 233,228 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-06-22 22:06:09 |
合計ジャッジ時間 | 6,333 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
#include "bits/stdc++.h"#include "atcoder/all"using namespace std;using namespace atcoder;//using mint = modint1000000007;//const int mod = 1000000007;using mint = modint998244353;const int mod = 998244353;//const int INF = 1e9;//const long long LINF = 1e18;//const bool debug = false;#define rep(i, n) for (int i = 0; i < (n); ++i)#define rep2(i,l,r)for(int i=(l);i<(r);++i)#define rrep(i, n) for (int i = (n-1); i >= 0; --i)#define rrep2(i,l,r)for(int i=(r-1);i>=(l);--i)#define all(x) (x).begin(),(x).end()#define allR(x) (x).rbegin(),(x).rend()#define endl "\n"#define P pair<int,int>template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }struct dat {int idx, a, x;bool operator< (const dat &right)const {return a * (right.x + 1) < right.a * (x + 1);}};mint calc(vector<int>&a, vector<int>&x, int &k) {mint ans = 1;rep(i, k)ans *= i + 1;rep(i, a.size()) {mint sub = 1;rep(j, x[i]) sub *= j + 1;ans *= sub.inv();}rep(i, a.size()) ans *= pow_mod(a[i], x[i], mod);{int sum = 0;rep(i, a.size()) sum += a[i];int pw = pow_mod(sum, k, mod);ans *= ((mint)pw).inv();}return ans;}int main() {ios::sync_with_stdio(false);cin.tie(nullptr);int t; cin >> t;while (t--) {int n, k; cin >> n >> k;vector<int>a(n), x(n);rep(i, n)cin >> a[i];priority_queue<dat> pq;rep(i, n)pq.push(dat{ i,a[i],x[i] });rep(i, k) {auto p = pq.top();pq.pop();x[p.idx]++;pq.push({ p.idx,a[p.idx],x[p.idx] });}cout << calc(a, x, k).val() << endl;}return 0;}