結果
問題 |
No.1011 Infinite Stairs
|
ユーザー |
![]() |
提出日時 | 2020-03-21 14:58:44 |
言語 | C++11(廃止可能性あり) (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 225 ms / 2,000 ms |
コード長 | 2,005 bytes |
コンパイル時間 | 1,403 ms |
コンパイル使用メモリ | 163,028 KB |
実行使用メモリ | 217,728 KB |
最終ジャッジ日時 | 2024-07-17 11:59:50 |
合計ジャッジ時間 | 3,117 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include "bits/stdc++.h" using namespace std; typedef long long ll; constexpr int MOD = 1e9 + 7; template<long long mod> struct mint { private: long long x; public: mint(long long x = 0) :x((mod + x) % mod) {} mint(std::string &s) { long long z = 0; for (int i = 0; i < s.size(); i++) { z *= 10; z += s[i] - '0'; z %= mod; } this->x = z; } 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) { long long n = mod - 2; mint<mod> u = 1, b = a; while (n > 0) { if (n & 1) { u *= b; } b *= b; n >>= 1; } return *this *= u; } 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 operator/(const mint &a) const { mint res(*this); return res /= a; } friend std::ostream& operator<<(std::ostream &os, const mint &n) { return os << n.x; } bool operator==(const mint &a) const { return this->x == a.x; } }; int main() { int n, d, k; cin >> n >> d >> k; // dp[i][j]:i段目につくまでj回行動する数 vector<vector<mint<MOD>>> dp(k + 1, vector<mint<MOD>>(n + 1, 0)); dp[0][0] = 1; for (int i = 0; i < k; i++) { for (int j = 0; j < n; j++) { dp[i + 1][j + 1] = dp[i][j + 1] + dp[i][j] - (i >= d ? dp[i - d][j] : 0); } } cout << dp[k][n] << endl; return 0; }