POJ 1976 A Mini Locomotive （动态规划）

哈哈这是一道典型的动态规划，呵呵有背包的思想在里面，但是还是要根据题目灵活运用。

先说说题目的意思吧：题目的大概意思就是说给你n个数，然后就是有三辆货车头可以拉连续k辆车厢，问你这三个火车头最终可以拉的最大的乘客数是多少。

我们首先定义dp的含义吧，dp[i][j]表示用i个火车头，在前j个车厢中可以拉到的最多的乘客数，那么转移方程可以很好的得到如下：

dp[i][j]=max(dp[i][j-1],dp[i-1][j-k]+b[j]);

其中的变量需要解释一下b[j]表示从j-k+1到j的k节车厢的总的人数，那么我们可以把连续的k节车厢看成一个物品，看是否使用，如果使用则是转移方程的前一项，否则就是不选该车厢，那么就是转移方程的后一项。这样定义出来了转移方程就很好写了：

下面是ac代码：

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#define Max 50010
int dp[4][Max];
int a[Max];
int b[Max];

int inline max(int a,int b)
{
    if(a>b)
        return a;
    else
        return b;
}

int solve(int n,int k)
{
    int i,j;
    int f;
    memset(dp,0,sizeof(dp));
    for(i=1;i<=3;i++)
    {
        for(j=k;j<=n;j++)
        {
            dp[i][j]=max(dp[i-1][j-k]+b[j],dp[i][j-1]);
        }
    }
    return dp[3][n];
}

int main()
{
    int T,n,sum,i,k;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        a[0]=0;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        scanf("%d",&k);
        sum=0;
        b[0]=a[0];
        for(i=1;i<=k;i++)
        {
            b[i]=b[i-1]+a[i];
        }
        sum=b[k];
        for(i=k+1;i<=n;i++)
        {
            sum+=a[i];
            sum-=a[i-k];
            b[i]=sum;
        }
        printf("%d\n",solve(n,k));
    }
    return 0;
}

这道题总的来说比较简单，理解转移的过程就可以轻松的写出转移方程了。