BZOJ3398: [Usaco2009 Feb]Bullcow 牡牛和牝牛(dp)

題意

    約翰要帶N(1≤N≤100000)只牛去參加集會裡的展示活動，這些牛可以是牡牛，也可以是牝牛．牛們要站成一排．但是牡牛是好鬥的，為了避免牡牛鬧出亂子，約翰決定任意兩隻牡牛之間至少要有K(O≤K<N)只牝牛．     請計算一共有多少種排隊的方法．所有牡牛可以看成是相同的，所有牝牛也一樣．答案對5000011取模

Sol

網上的題解是首碼和優化dp？

那我說一個不一樣的做法

設$f[i]$表示到第$i$個位置，該位置放了牡牛的方案，$g[i]$表示到第$i$個位置，且該位置放了牝牛的方案數

然後兩個數組可以互相推出來

#include<cstdio>
#include<algorithm>
#include<stack>
#include<queue>
#include<cmath>
#define LL long long 
#define lb(x) (x & (-x))
#define Pair pair<int, int> 
#define fi first
#define se second
#define MP(x, y) make_pair(x, y)
using namespace std;
const int MAXN = 1e6 + 10, mod = 5000011;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar(); 
    return x * f;
}
int N, K;
int f[MAXN], g[MAXN];
int main() {
    N = read(); K = read();
    f[1] = 1; g[1] = 1;
    for(int i = 2; i <= N; i++) {
        f[i] = (f[i - 1] + g[i - 1]) % mod;
        g[i] = (f[max(i - K - 1, 1)] + g[max(i - K - 1, 0)]) % mod;
    }
    printf("%d", (f[N] + g[N]) % mod);
    return 0;
}
/*

*/