poj 1976 A Mini Locomotive(二维dp)

#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
#define MAX(a,b) (a>b?a:b)
const int M(51000);
int a[M],val[M],f[2][M];
void dp(int n,int m)
{
int i,maxm[3];
//memset(f,0,sizeof(f));
maxm[0]=val[0];
maxm[1]=0;
for(i=m;i<n;i++)
{
f[0][i]=maxm[0]+val[i];
f[1][i]=maxm[1]+val[i];
maxm[0]=MAX(val[i-m+1],maxm[0]);
maxm[1]=MAX(f[0][i-m+1],maxm[1]);
//f[i]=val[i]+maxm;
//maxm=MAX(f[i-m+1],maxm);
}
maxm[0]=0;
for(i=0;i<n;i++)
maxm[0]=MAX(f[1][i],maxm[0]);
printf("%d\n",maxm[0]);
}
int main()
{
int t,i,j,k,n,m,sum;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(i=0;i<n;i++)
scanf("%d",&a[i]);
scanf("%d",&m);
sum=0;
for(i=0,j=0;i<n;i++)
{
sum+=a[i];
if(i>=m-1)
{
val[j++]=sum;
sum-=a[i-m+1];
}
}
dp(j,m);
}
}