P2690 接苹果

P2690 接苹果
f[i][j][k]表示
i=1或2，表示j时刻cow在哪棵树下
j表示时刻
k表示转移了k次
如果当前第1棵树落苹果
f[1][t][left]=max(f[1][t-1][left],f[2][t-1][left-1])+1;
如果当前第2棵树落苹果
f[2][t][left]=max(f[2][t-1][left],f[1][t-1][left-1])+1;

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cstring>
#define inf 2147483647
#define For(i,a,b) for(register int i=a;i<=b;i++)
#define p(a) putchar(a)
#define g() getchar()
//by war
//2017.10.18
using namespace std;
int n;
int x;
int w;
int f[10][1010][40];
int ans;
void in(int &x)
{
    int y=1;
    char c=g();x=0;
    while(c<'0'||c>'9')
    {
    if(c=='-')
    y=-1;
    c=g();
    }
    while(c<='9'&&c>='0')x=x*10+c-'0',c=g();
    x*=y;
}
void o(int x)
{
    if(x<0)
    {
        p('-');
        x=-x;
    }
    if(x>9)o(x/10);
    p(x%10+'0');
}

void zy(int t)
{
        f[1][t][0]++;   
        for(int left=2;left<=w;left+=2)
        f[1][t][left]=max(f[1][t-1][left],f[2][t-1][left-1])+1;
}

void yz(int t)
{
        for(int left=1;left<=w;left+=2)
        f[2][t][left]=max(f[2][t-1][left],f[1][t-1][left-1])+1;
}

int main()
{
    in(n),in(w);
    For(t,1,n)
    {
        for(int left=1;left<=w;left+=2)
        f[2][t][left]=f[2][t-1][left];
        for(int left=0;left<=w;left+=2)
        f[1][t][left]=f[1][t-1][left];
        in(x);
        if(x==1)
        zy(t);
        else
        yz(t);
    }
    for(int i=1;i<=w;i+=2)
    ans=max(ans,f[2][n][i]);
    for(int i=0;i<=w;i+=2)
    ans=max(ans,f[1][n][i]);
    o(ans);
     return 0;
}