描述
很少有人知道奶牛愛喫蘋果。農夫約翰的農場上有兩棵蘋果樹(編號爲1和2), 每一棵樹上都長滿了蘋果。奶牛貝茜無法摘下樹上的蘋果,所以她只能等待蘋果 從樹上落下。但是,由於蘋果掉到地上會摔爛,貝茜必須在半空中接住蘋果(沒有人愛喫摔爛的蘋果)。貝茜喫東西很快,她接到蘋果後僅用幾秒鐘就能喫完。每一分鐘,兩棵蘋果樹其中的一棵會掉落一個蘋果。貝茜已經過了足夠的訓練, 只要站在樹下就一定能接住這棵樹上掉落的蘋果。同時,貝茜能夠在兩棵樹之間 快速移動(移動時間遠少於1分鐘),因此當蘋果掉落時,她必定站在兩棵樹其中的一棵下面。此外,奶牛不願意不停地往返於兩棵樹之間,因此會錯過一些蘋果。蘋果每分鐘掉落一個,共T(1<=T<=1000)分鐘,貝茜最多願意移動W(1<=W<=30) 次。現給出每分鐘掉落蘋果的樹的編號,要求判定貝茜能夠接住的最多蘋果數。 開始時貝茜在1號樹下。
輸入
第一行2個數,t和k。接下來的t行,每行一個數,代表在時刻t蘋果是從1號蘋果樹還是從2號蘋果樹上掉下來的。
輸出
對於每個測試點,輸出一行,一個數,爲奶牛最多接到的蘋果的數量。
樣例輸入
7 2
2
1
1
2
2
1
1
樣例輸出
6
提示
輸出註解:貝茜不移動直到接到從第1棵樹上掉落的兩個蘋果,然後移動到第2棵樹下,直到接到從第2棵樹上掉落的兩個蘋果,最後移動到第1棵樹下,接住最後兩個從第1 棵樹上掉落的蘋果。這樣貝茜共接住6個蘋果。
動態規劃問題,f(i,j)爲第i分鐘移動j次喫到的蘋果數,f(i,j)=max(f(i-1,j-1),f(i-1.j-1))。可以建立一個T*W的矩陣,根據狀態轉移方程爲矩陣賦值,遍歷最後一行就可得到最大值(即矩陣第t行)
#include <iostream>
using namespace std;
#define MAX(a,b)a>b?a:b
/*int MAX(int a,int b)
{
return a>b?a:b;
}*/
int _tmain(int argc, _TCHAR* argv[])
{
int s[1001][31];
int max=0,k,t,i,j,x,y;
int a[1001];
cin>>t>>k;
for(i=1;i<=t;++i)
cin>>a[i];
for(i=0;i<=k;++i)
s[0][i]=0;//時間t=0時喫到的蘋果數必爲0
for(i=1;i<=t;++i)
{
s[i][0]=s[i-1][0]+a[i]%2;//移動次數爲0時
for(j=1;j<=k;++j)
{
x=(j%2==(a[i]-1)?1:0);//計算這一次能否喫到蘋果
y=MAX(s[i-1][j-1],s[i-1][j]);
s[i][j]=x+y;
}
}
for(i=0;i<=k;++i)
if(s[t][i]>max) max=s[t][i];
cout<<max;
return 0;
}