BZOJ 3398 [Usaco2009 Feb]Bullcow 牡牛和牝牛：dp【字首和優化】

題目連結：http://www.lydsy.com/JudgeOnline/problem.php?id=3398

題意：

　　約翰要帶N(1≤N≤100000)只牛去參加集會裡的展示活動，這些牛可以是牡牛，也可以是牝牛。

　　牛們要站成一排。但是牡牛是好鬥的，為了避免牡牛鬧出亂子，約翰決定任意兩隻牡牛之間至少要有K(0≤K<N)只牝牛。

　　請計算一共有多少種排隊的方法。所有牡牛可以看成是相同的，所有牝牛也一樣。答案對5000011取模。

 

題解：

　　表示狀態：

　　　　dp[i] = num of ways

　　　　表示考慮到位置i，並且在這裡放了牡牛的方案數。

 

　　找出答案：

　　　　ans = ∑(dp[i]) + 1

　　　　因為不放牡牛也算一種方案，所以最後+1。

 

　　如何轉移：

　　　　dp[i] = ∑ dp[0 to i-k-1] + 1

　　　　上一次放牡牛的位置至少在i前面k+1個牛的位置。

　　　　或者這是第一次放牡牛，所以+1。

 

　　優化：

　　　　字首和優化。

 

AC Code:

// state expression:
// dp[i] = num of ways
// i: last pos of cow2
//
// find the answer:
// sigma dp[i] + 1
//
// transferring:
// dp[i] = sigma dp[0 to i-k-1] + 1
//
// boundary:
// set dp = 0
#include <iostream>
#include <stdio.h>
#include <string.h>
#define MAX_N 100005
#define MOD 5000011

using namespace std;

int n,k;
int ans;
int dp;
int sum[MAX_N];

void read()
{
    cin>>n>>k;
}

void solve()
{
    sum[0]=0;
    ans=1;
    for(int i=1;i<=n;i++)
    {
        dp=1;
        if(i-k-1>=0) dp=(dp+sum[i-k-1])%MOD;
        sum[i]=(sum[i-1]+dp)%MOD;
        ans=(ans+dp)%MOD;
    }
}

void print()
{
    cout<<ans<<endl;
}

int main()
{
    read();
    solve();
    print();
}