結果

問題 No.1011 Infinite Stairs
ユーザー shibh308shibh308
提出日時 2020-03-20 21:33:28
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,334 ms / 2,000 ms
コード長 2,572 bytes
コンパイル時間 2,091 ms
コンパイル使用メモリ 206,504 KB
実行使用メモリ 215,808 KB
最終ジャッジ日時 2024-07-17 11:44:58
合計ジャッジ時間 7,573 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 12 ms
6,940 KB
testcase_03 AC 812 ms
143,232 KB
testcase_04 AC 124 ms
26,880 KB
testcase_05 AC 1,334 ms
215,808 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 8 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 6 ms
6,940 KB
testcase_11 AC 148 ms
31,744 KB
testcase_12 AC 5 ms
6,944 KB
testcase_13 AC 90 ms
20,736 KB
testcase_14 AC 2 ms
6,940 KB
testcase_15 AC 114 ms
25,984 KB
testcase_16 AC 558 ms
101,120 KB
testcase_17 AC 34 ms
11,392 KB
testcase_18 AC 325 ms
59,776 KB
testcase_19 AC 5 ms
6,940 KB
testcase_20 AC 3 ms
6,940 KB
testcase_21 AC 127 ms
26,240 KB
testcase_22 AC 256 ms
49,408 KB
testcase_23 AC 155 ms
30,848 KB
testcase_24 AC 169 ms
33,536 KB
testcase_25 AC 263 ms
52,608 KB
testcase_26 AC 55 ms
13,184 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;

using i64 = long long;

const i64 MOD = 1e9 + 7;
const i64 INF = i64(1e18) + 7;

template <typename T>
bool chmin(T& x, T y){
    if(x > y){
        x = y;
        return true;
    }
    return false;
}

template <typename T>
bool chmax(T& x, T y){
    if(x < y){
        x = y;
        return true;
    }
    return false;
}

template <i64 mod = MOD>
struct ModInt{
    i64 p;

    ModInt() : p(0){}
    ModInt(i64 x){p = x >= 0 ? x % mod : x + (-x + mod - 1) / mod * mod;}

    ModInt& operator+=(const ModInt& y){p = p + *y - ((p + *y) >= mod ? mod : 0); return *this;}
    ModInt& operator-=(const ModInt& y){p = p - *y + (p - *y < 0 ? mod : 0); return *this;}
    ModInt& operator*=(const ModInt& y){p = (p * *y) % mod; return *this;}
    ModInt& operator%=(const ModInt& y){if(y)p %= *y; return *this;}

    ModInt operator+(const ModInt& y) const{ModInt x = *this; return x += y;}
    ModInt operator-(const ModInt& y) const{ModInt x = *this; return x -= y;}
    ModInt operator*(const ModInt& y) const{ModInt x = *this; return x *= y;}
    ModInt operator%(const ModInt& y) const{ModInt x = *this; return x %= y;}

    friend ostream& operator<<(ostream& stream, const ModInt<mod>& x){
        stream << *x;
        return stream;
    }

    friend ostream& operator>>(ostream& stream, const ModInt<mod>& x){
        stream >> *x;
        return stream;
    }

    ModInt& operator++(){p = (p + 1) % mod; return *this;}
    ModInt& operator--(){p = (p - 1 + mod) % mod; return *this;}

    bool operator==(const ModInt& y) const{return p == *y;}
    bool operator!=(const ModInt& y) const{return p != *y;}

    const i64& operator*() const{return p;}
    i64& operator*(){return p;}

};

using mint = ModInt<>;


template <typename T>
struct BIT{
    vector<T> elm;
    BIT(int n, T init = T()) : elm(n + 1, init){
    }

    // [0, x)
    T sum(int x){
        T val = 0;
        for(; x > 0; x -= x & -x)
            val += elm[x];
        return val;
    }

    // [l, r)
    T sum(int l, int r){
        return sum(r) - sum(l);
    }

    void add(int x, T val){
        for(++x; x < elm.size(); x += x & -x)
            elm[x] += val;
    }
};


signed main(){
    int n, d, k;
    cin >> n >> d >> k;
    // セグ木じゃ通らないんかーい!
    vector<BIT<mint>> dp(n + 1, BIT<mint>(k + 1));
    dp[0].add(0, 1);
    for(int i = 0; i < n; ++i)
        for(int j = 0; j <= k; ++j)
            dp[i + 1].add(j, dp[i].sum(max(0, j - d), j));
    cout << dp.back().sum(k, k + 1) << endl;
    return 0;
}
0