HZAU校赛补题PA(动态规划)

说来惭愧啊，那么简单的动态规划签到题我当时居然没把状态转移方程给推出来

搞得自己变成三队的了

题目传送门http://acm.hzau.edu.cn/problem.php?id=1199

这题目要注意的地方我就放在注释里了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 100000+100

using namespace std;

int a[maxn],dp[maxn][3];

int main()
{
    int t,n,k;
    scanf("%d",&t);
    while(t--)
    {
        int max1,max0;
        scanf("%d %d",&n,&k);
        memset(a,0,sizeof(a));
        memset(dp,0,sizeof(dp));
        for (int i=1;i<=n;i++)
            scanf("%d",&a[i]);

        max0=0;max1=0;
        for (int i=1;i<=n;i++)
        {
            dp[i][0]=max0;
            dp[i][1]=a[i]+max1;

            max0=max(dp[i][0],dp[i][1]);
            if (i-k>=1)
                max1=max(max1,max(dp[i-k][0],dp[i-k][1]));
        }
        printf("%d\n",max(dp[n][0],dp[n][1]));
    }
    return 0;
}

/*1 2 3 4 5 6

对于下一个的max0和max1
i=3
k=1
取2*/