結果
問題 | No.1693 Invasion |
ユーザー |
|
提出日時 | 2021-10-01 22:15:08 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 137 ms / 2,000 ms |
コード長 | 1,374 bytes |
コンパイル時間 | 1,660 ms |
コンパイル使用メモリ | 178,480 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-19 11:35:53 |
合計ジャッジ時間 | 3,380 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h>#define rep(i, l, r) for (int i = (l); i < (r); i++)using namespace std;typedef long long ll;ll MOD = 998244353;vector<ll> f(1e5 + 1, 1);ll mod_pow(ll x, ll n, ll mod) {ll res = 1;while (n > 0) {if (n & 1) res = res * x % mod;x = x * x % mod;n >>= 1;}return res;}ll C(ll n, ll k) {ll ret = f[n] * mod_pow(f[k], MOD - 2, MOD) % MOD;ret = ret * mod_pow(f[n - k], MOD -2, MOD) % MOD;return ret;}int main() {ll ans = 0;int N, M;cin >> N >> M;rep(i, 0, M) f[i + 1] = f[i] * (i + 1) % MOD;vector<int> A(N);rep(i, 0, N) cin >> A[i];queue<pair<int, int>> q;vector<int> b(M + 1, 1e9); //x個選ぶのに最小でb(x)回必要;q.push(make_pair(0, 0));map<pair<int, int >, bool> done;b[0] = 0;while (!q.empty()) {int x = q.front().first, y = q.front().second;//cout << "(" << x << "," << y << ")";q.pop();rep(i, 0, N) {if (y + A[i] <= M && x + 1 < b[y + A[i]]) {b[y + A[i]] = x + 1;q.push(make_pair(x + 1, y + A[i]));}}}rep(i, 0, M + 1) {//cout << i << " " << b[i] << endl;if (b[i] == 1e9) continue;ans += C(M - b[i], i - b[i]);ans %= MOD;}cout << ans << endl;}