#484701 (C++17) No.1172 Add Recursive Sequence

• 問題: No.1172 Add Recursive Sequence
• ユーザー: opt
• 提出日時: 2020-05-22 15:14:26
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 723 ms / 4,000 ms
• コード長: 2,005 bytes
• コンパイル時間: 3,076 ms
• コンパイル使用メモリ: 201,504 KB
• 実行使用メモリ: 165,760 KB
• 最終ジャッジ日時: 2024-04-17 06:59:10
• 合計ジャッジ時間: 4,608 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	3 ms; 5,376 KB
testcase_07	AC	3 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	10 ms; 5,504 KB
testcase_11	AC	10 ms; 5,760 KB
testcase_12	AC	10 ms; 5,504 KB
testcase_13	AC	10 ms; 5,376 KB
testcase_14	AC	66 ms; 17,280 KB
testcase_15	AC	51 ms; 17,408 KB
testcase_16	AC	723 ms; 165,720 KB
testcase_17	AC	425 ms; 165,760 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
#define rep2(i, m, n) for(int i=int(m); i<int(n); ++i)
#define rep(i, n) rep2(i, 0, n)
using ll = long long;
struct fast_ios { fast_ios(){ cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_;
const int MOD = int(1e9) + 7;

struct mint {
  ll x;
  mint(ll x=0) : x((x+MOD)%MOD) {}
  mint operator-() const { return mint(-x);}
  mint& operator+=(const mint a) { if ((x += a.x) >= MOD) x -= MOD; return *this; }
  mint& operator-=(const mint a) { if ((x -= a.x) < 0) x += MOD; return *this; }
  mint& operator*=(const mint a) { (x *= a.x) %= MOD; return *this;}
  mint operator+(const mint a) const { return mint(*this) += a;}
  mint operator-(const mint a) const { return mint(*this) -= a;}
  mint operator*(const mint a) const { return mint(*this) *= a;}
};
istream& operator>>(istream& is, mint& a) { return is >> a.x;}
ostream& operator<<(ostream& os, const mint& a) { return os << a.x;}
using Vm = vector<mint>;
using VVm = vector<Vm>;


int main() {
  int k, n, m; cin >> k >> n >> m;
// assert //
  assert(1 <= k && k <= 200);
  assert(1 <= n && n <= int(1e5));
  assert(1 <= m && m <= int(1e5));
////////////

  Vm a(k+n), c(k+1);
  c[0] = -1;
  rep(i, k) {
    int x; cin >> x;
// assert //
    assert(0 <= x && x <= int(1e9));
////////////
    a[i] = x;
  }
  rep(i, k) {
    int x; cin >> x;
// assert //
    assert(0 <= x && x <= int(1e9));
////////////
    c[i+1] = x;
  }

  rep2(i, k, n) rep2(j, 1, k+1) a[i] += c[j]*a[i-j];

  VVm d(n, Vm(k));
  rep(i, n) {
    d[i][0] = c[0]*a[i];
    rep2(j, 1, k) d[i][j] += d[i][j-1] + c[j]*a[i-j];
  }

  Vm y(n);
  rep(_, m) {
    int l, r; cin >> l >> r;
// assert //
    assert(0 <= l && l < r && r <= n);
////////////
    rep2(i, l, min(l+k, n)) y[i] += d[i-l][i-l];
    rep2(i, r, min(r+k, n)) y[i] -= d[i-l][i-r];
  }

  Vm x(n);
  rep(i, n) {
    x[i] = - y[i];
    rep2(j, 1, min(i, k)+1) x[i] += c[j]*x[i-j];
    cout << x[i] << "\n";
  }
  return 0;
}