跳舞_纪中2545_dp

Description

　　小明今天得到一个跳舞毯游戏程序Dance。游戏每次连续出N 个移动的“箭头”，箭头依次标号为1 到N，并且的相应的分数S[1..N]。如果你能“踏中”第i 号箭头，你将获得相应的分数S[i]；否则将被扣除相应的分数。
　　另外，游戏还有一个累计奖励机制：如果踏准次数累计达到T，并且是在踏中第i个箭头达到的，则将得到B[i]的奖励分数,累计也将清零，重新开始。
　　例如：N＝6，T=3，相应的S 和B 分别为{1,2,3,4,5,6}、{0,0,4,7,9,10},如果小明踏中所有箭头，则得分为：(1+2+3+4)+(4+5+6+10)＝35
　　小明是个Dance 高手，可以踏中他想踏中的任意一个箭头。但他发现，根据给定的N，T，S，B，踏中所有的箭头不一定能得最高分，小明很想知道最高能得多少分，你能帮助小明计算一下最多可得多少分吗？

Input

　　第一行两个整数N 和T。
　　第二行N 个整数，为S 的相应分数。
　　第三行也有N 个整数，为B 的相应分数。

Output

　　只一个整数，可得到的最高分数。

Hint

[数据范围]

　　对于20%的数据0≤N,T≤100；
　　对于100%的数据0≤N,T≤5000；
　　S 和B 各有N 个数，所有分数为[0，10000]之间的整数。

题解

$f[i][j]$表示前i项共选了j项的最大获益

$f[i][j]=\max(f[i-1][j]-s[i],f[i-1][j-1]+s[i])$

当j为t的倍数

$f[i][j]=\max(f[i-1][j]-s[i],f[i-1][j-1]+s[i]+b[i])$

固定 $f[i][0]$ 只能从 $f[i-1][0]$ 转移

i可以滚，所以数组很小

code

#include <stdio.h>
#include <cstring>
using namespace std;
int max(int x,int y)
{
    return x>y?x:y;
}
long long f[2][5100];
int a[5100];
int b[5100];
int main()
{
    int n,m;
    long long ans=0;
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for (int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    int x=1;
    for (int i=1;i<=n;i++)
    {
        f[x][0]=f[x^1][0]-a[i];
        for (int j=1;j<=i;j++)
        {
            if (j%m)
                f[x][j]=max(f[x^1][j]-a[i],f[x^1][j-1]+a[i]);
            else
                f[x][j]=max(f[x^1][j]-a[i],f[x^1][j-1]+a[i]+b[i]);
        }
        x=x^1;
    }
    for (int i=0;i<=n;i++)
        if (f[x^1][i]>ans)
            ans=f[x^1][i];
    printf("%lld\n",ans);
    return 0;
}