結果
問題 |
No.1011 Infinite Stairs
|
ユーザー |
|
提出日時 | 2020-09-16 23:53:11 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,455 ms / 2,000 ms |
コード長 | 1,215 bytes |
コンパイル時間 | 2,142 ms |
コンパイル使用メモリ | 195,692 KB |
最終ジャッジ日時 | 2025-01-14 15:28:37 |
ジャッジサーバーID (参考情報) |
judge1 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define REP(i,n) for(int i=0; i<(int)(n); i++) const long long MOD = 1e9 + 7; template <typename T> class BIT { vector<T> bit; public: BIT() {} // manage data in [1, n] BIT(int n) : bit(n + 1) {} void init() { fill(bit.begin(), bit.end(), 0); } // return sum in [1, i] T sum(int i){ T s = 0; while(i > 0){ s += bit[i]; if (s >= MOD) s -= MOD; i -= i & -i; } return s; } // return sum in [l, r] T sum(int l, int r) { long long ret = sum(r) - sum(l-1); ret %= MOD; if (ret < 0) ret += MOD; return ret; } void add(int i, T x){ while(i < (int)bit.size()){ bit[i] += x; if (bit[i] >= MOD) bit[i] -= MOD; i += i & -i; } } }; int n, d, k; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin >> n >> d >> k; BIT<long long> dp(k); --n; for (int i = 1; i <= d; i++) dp.add(i, 1); REP (i, n) { for (int j = k; j >= 1; j--) { long long cur = dp.sum(j, j); long long nxt = dp.sum(max(1, j-d), j-1); dp.add(j, nxt-cur); } } cout << dp.sum(k, k) << endl; return 0; }