Submission #1797385

---

Source Code Expand

#define MAX_N 1000000
#define MAX_K 1000000
#define mod(x, y) ((x % y) + y) % y
#define MOD 1000000007
#define modp(x) mod(x, MOD)

int main(int argc, const char * argv[]) {
    // a[i] : 駅1から駅iまでの停まり方(駅iには必ず止まる)の総数
    ll a[MAX_N];
    // b[i] : 駅1から駅iまでの停まり方(駅iには止まらない)の総数
    ll b[MAX_N];
    
    int N, K;
    cin >> N >> K;
    
    a[1] = 1;
    b[1] = 0;
    for (int i = 1; i < K-1; i++) {
        a[i+1] = modp(a[i] + b[i]);
        b[i+1] = modp(a[i] + b[i]);
    }
    
    a[K] = modp(a[K-1] + b[K-1] - 1);
    b[K] = modp(a[K-1] + b[K-1]);
    
    for (int i = K; i < N; i++) {
        a[i+1] = modp((a[i] - b[i-(K-1)]) + b[i]);
        b[i+1] = modp((a[i] - b[i-(K-1)]) + b[i] + b[i-(K-1)]);
    }
    cout << a[N] << endl;
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main(int, const char**)’:
./Main.cpp:9:5: error: ‘ll’ was not declared in this scope
     ll a[MAX_N];
     ^
./Main.cpp:11:8: error: expected ‘;’ before ‘b’
     ll b[MAX_N];
        ^
./Main.cpp:14:5: error: ‘cin’ was not declared in this scope
     cin >> N >> K;
     ^
./Main.cpp:16:5: error: ‘a’ was not declared in this scope
     a[1] = 1;
     ^
./Main.cpp:17:5: error: ‘b’ was not declared in this scope
     b[1] = 0;
     ^
./Main.cpp:30:5: error: ‘cout’ was not declared in this scope
     cout << a[N] << endl;
     ^
./Main.cpp:30:21: error: ‘endl’ was not declared in this scope
     cout << a[N] << endl;
                     ^