結果
問題 |
No.1353 Limited Sequence
|
ユーザー |
|
提出日時 | 2021-05-05 18:21:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 197 ms / 2,000 ms |
コード長 | 2,658 bytes |
コンパイル時間 | 2,012 ms |
コンパイル使用メモリ | 200,520 KB |
最終ジャッジ日時 | 2025-01-21 07:34:00 |
ジャッジサーバーID (参考情報) |
judge4 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 46 |
ソースコード
#include <bits/stdc++.h> using namespace std; template<typename T,T MOD> struct mod_int{ T v; mod_int(): v(0) {} mod_int(T t) { v=t%MOD; if(v<0) v+=MOD; } mod_int inv() const { T a=v, b=MOD, u=1, v=0; while (b>0) { T t=a/b; a-=t*b; swap(a,b); u-=t*v; swap(u,v); } return u; } mod_int pow(long long p) const { mod_int ret = 1, base(v); while (p > 0) { if (p & 1) ret *= base; base *= base; p >>= 1; } return ret; } mod_int& operator+=(const mod_int &a) { v+=a.v; if(v>=MOD)v-=MOD; return *this; } mod_int& operator-=(const mod_int &a) { v+=MOD-a.v; if(v>=MOD)v-=MOD; return *this; } mod_int& operator*=(const mod_int &a) { v=v*a.v%MOD; return *this; } mod_int& operator/=(const mod_int &a) { return (*this)*=a.inv(); } mod_int operator+(const mod_int &a) const { return mod_int(v)+=a; } mod_int operator-(const mod_int &a) const { return mod_int(v)-=a; } mod_int operator*(const mod_int &a) const { return mod_int(v)*=a; } mod_int operator/(const mod_int &a) const { return mod_int(v)/=a; } friend mod_int operator+(T x, const mod_int& m) { return mod_int(x)+m; } friend mod_int operator-(T x, const mod_int& m) { return mod_int(x)-m; } friend mod_int operator*(T x, const mod_int& m) { return mod_int(x)*m; } friend mod_int operator/(T x, const mod_int& m) { return mod_int(x)/m; } mod_int operator-() const{ return v?mod_int(MOD-v):mod_int(v); } bool operator==(const mod_int &a) const { return v==a.v; } bool operator!=(const mod_int &a) const { return v!=a.v; } bool operator<(const mod_int &a) const { return v <a.v; } friend istream& operator>>(istream& is, mod_int& m){ T t; is>>t; m=mod_int(t); return is; } friend ostream& operator<<(ostream& os,const mod_int& m){ return os<<m.v; } }; using mint = mod_int<long long, 998244353>; int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n, l, r; cin >> n >> l >> r; vector<int> a(n + 1); // a[i] = 連続してiを使用できる上限値 for (int i = 1; i <= n; i++) cin >> a[i]; vector<vector<mint> > dp(r + 1, vector<mint>(n + 1)); // dp[要素の和][最後に使った数] = Bのパターン数 dp[0][0] = 1; for (int i = 0; i < r; i++) { mint sum = accumulate(dp[i].begin(), dp[i].end(), mint(0)); for (int j = 1; j <= n; j++) { int s = i; for (int k = 0; k < a[j]; k++) { s += j; if (s > r) break; dp[s][j] += sum - dp[i][j]; } } } mint ret = 0; for (int i = l; i <= r; i++) ret += accumulate(dp[i].begin(), dp[i].end(), mint(0)); cout << ret << endl; return 0; }