poj-1976 01背包

A Mini Locomotive

     题意：输入第一个数为测试数据的数量。输入第二个数n表示有N节车厢。输入第三个有n个数，表示第1个车厢到第n个车厢各个车厢的乘客数量。输入第三个数m表示一个火车头可以拉几个车厢，每个火车头拉走的车厢都是连续的。一共有三个火车头，求三个火车头最多可以拉走多少乘客。

     这题像一个背包问题，求要取哪些使得价值最大。但是又略有改变，要求取走的是连续的，需要做些处理将题目变成真正的01背包问题。我们可以把连续的m个车厢里的乘客累加起来当做一个车厢处理，对应于背包问题来说，原来每个车厢体积为1，转化成了体积为m。

#include<iostream>
#include<cstring>
#include<math.h>
using namespace std;
int a[50010],b[50010];
int dp[4][50010];
int main()
{
	int t,n,m;
	cin >> t;
	while(t--)
	{
		cin >> n;
		for(int i=1;i<=n;i++)
		{
			cin >> a[i];
		}
		cin >> m;
		memset(b,0,sizeof(b));
		for(int i=m;i<=n;i++)
		{
			for(int j=0;j<m;j++)
			{
				b[i] += a[i-j];
			}
		}
		memset(dp,0,sizeof(dp));
		
		for(int i=1;i<=3;i++)
		{
			for(int j=m*i;j<=n;j++)
			{
				dp[i][j] = max(dp[i-1][j-m]+b[j],dp[i][j-1]);
			}
		}
		cout << dp[3][n] << endl;
	}
	return 0;
}