Problem p02549 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); ++i)
using namespace std;
using ll = long long;
const int INF = 1001001001;
const int mod = 998244353;
struct mint {
  // 自動的にmodを取ってくれるクラスです。
  ll x;
  mint(ll x = 0) : x(x % mod) {}
  mint &operator+=(const mint a) {
    if ((x += a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator-=(const mint a) {
    if ((x += mod - a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator*=(const mint a) {
    (x *= a.x) %= mod;
    return *this;
  }
  mint operator+(const mint a) const {
    mint res(*this);
    return res += a;
  }
  mint operator-(const mint a) const {
    mint res(*this);
    return res -= a;
  }

  mint operator*(const mint a) const {
    mint res(*this);
    return res *= a;
  }
  mint pow(ll t) const {
    if (!t)
      return 1;
    mint b = pow(t / 2);
    b *= b;
    if (t % 2 == 1)
      b *= *this;
    return b;
  }

  mint inv() const { return pow(mod - 2); }
  mint &operator/=(const mint a) {
    (*this) *= a.inv();
    return *this;
  }
  mint operator/(const mint a) const {
    mint res(*this);
    return res /= a;
  }
};

int main() {
  int n, k;
  cin >> n >> k;
  vector<mint> dp(n + 1);
  vector<mint> dpsum(n + 1);
  vector<int> l(k), r(k);
  rep(i, k) cin >> l.at(i) >> r.at(i);
  dp.at(1) = mint(1);
  dpsum.at(1) = mint(1);
  for (int i = 2; i <= n; i++) {
    rep(j, k) {
      int li = i - r.at(j);
      int ri = i - l.at(j);
      if (ri < 0)
        continue;
      li = max(li, 1);
      dp.at(i) += dpsum.at(ri) - dpsum.at(li - 1);
    }
    dpsum.at(i) = dpsum.at(i - 1) + dp.at(i);
  }
  cout << dp.at(n).x << endl;
}

Error: TLE

Fixed submission:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (n); ++i)
using namespace std;
using ll = long long;
const int INF = 1001001001;
const int mod = 998244353;
struct mint {
  // 自動的にmodを取ってくれるクラスです。
  ll x;
  mint(ll x = 0) : x(x % mod) {}
  mint &operator+=(const mint a) {
    if ((x += a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator-=(const mint a) {
    if ((x += mod - a.x) >= mod)
      x -= mod;
    return *this;
  }
  mint &operator*=(const mint a) {
    (x *= a.x) %= mod;
    return *this;
  }
  mint operator+(const mint a) const {
    mint res(*this);
    return res += a;
  }
  mint operator-(const mint a) const {
    mint res(*this);
    return res -= a;
  }

  mint operator*(const mint a) const {
    mint res(*this);
    return res *= a;
  }
  mint pow(ll t) const {
    if (!t)
      return 1;
    mint b = pow(t / 2);
    b *= b;
    if (t % 2 == 1)
      b *= *this;
    return b;
  }

  mint inv() const { return pow(mod - 2); }
  mint &operator/=(const mint a) {
    (*this) *= a.inv();
    return *this;
  }
  mint operator/(const mint a) const {
    mint res(*this);
    return res /= a;
  }
};

int main() {
  int n, k;
  cin >> n >> k;
  set<int> s;
  vector<mint> dp(n);
  rep(i, k) {
    int l, r;
    cin >> l >> r;
    for (int j = l; j <= r; j++)
      s.insert(j);
  }
  // vector<int> sv;
  // for(auto itr = s.begin(); itr != s.end();itr++){
  //     int t = *itr;
  //     cout << t << endl;
  //     // sv.push_back(t);
  // }
  // sort(sv.begin(),sv.end());
  dp.at(0) = mint(1);
  rep(i, n) {
    for (auto itr = s.begin(); itr != s.end(); itr++) {
      int t = *itr;
      if ((i - t) < 0)
        continue;
      dp.at(i) += dp.at(i - t);
    }
  }
  cout << dp.at(n - 1).x << endl;
}