結果

問題 No.1353 Limited Sequence
ユーザー k
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
}
0