洛谷 P2690 接苹果

题目传送门

解题思路:

f[i][j]表示当前在第i个时间用了j此移动了,a[i][j]表示第i个时间j号树的苹果下落数.

如果移动了单数次,则在2号树,否则在1号树.

对于每个状态,我可以在上一个时间从另一棵树过来,也可以是我没动.

AC代码:

#include<iostream>
#include<cstdio>

using namespace std;

int t,n,f[1001][31],a[1001][2],ans;

inline int max(int s,int d) {
    if(s >= d) return s;
    return d;
}

int main() {
    scanf("%d%d",&n,&t);
    for(int i = 1;i <= n; i++) {
        int u;
        scanf("%d",&u);
        a[i][u-1] = 1;
    }
    for(int i = 1;i <= n; i++)
        for(int j = 0;j <= min(i,t); j++)
            if(j % 2 == 1)
                f[i][j] = max(f[i-1][j],f[i-1][j-1]) + a[i][1];
            else
                f[i][j] = max(f[i-1][j],f[i-1][j-1]) + a[i][0];
    for(int i = 0;i <= t; i++)
        ans = max(ans,f[n][i]);
    printf("%d",ans);
    return 0;
} 

然后,我发现可以压一维.

#include<iostream>
#include<cstdio>

using namespace std;

int t,n,f[31],a[1001][2],ans;

inline int max(int s,int d) {
    if(s >= d) return s;
    return d;
}

int main() {
    scanf("%d%d",&n,&t);
    for(int i = 1;i <= n; i++) {
        int u;
        scanf("%d",&u);
        a[i][u-1] = 1;
    }
    for(int i = 1;i <= n; i++)
        for(int j = 0;j <= min(i,t); j++)
            if(j % 2 == 1)//在2号树 
                f[j] = max(f[j],f[j-1]) + a[i][1];
            else//在一号树 
                f[j] = max(f[j],f[j-1]) + a[i][0];
    for(int i = 0;i <= t; i++)
        ans = max(ans,f[i]);
    printf("%d",ans);
    return 0;
}