The new ITone 6 has been released recently and George got really keen to buy it. Unfortunately, he didn't have enough money, so George was going to work as a programmer. Now he faced the following problem at the work.
Given a sequence of n integers p1, p2, ..., pn. You are to choose k pairs of integers:
in such a way that the value of sum is maximal possible. Help George to cope with the task.
The first line contains three integers n, m and k (1 ≤ (m × k) ≤ n ≤ 5000). The second line contains n integers p1, p2, ..., pn (0 ≤ pi ≤ 109).
Print an integer in a single line — the maximum possible value of sum.
5 2 1 1 2 3 4 5
9
7 1 3 2 10 7 18 5 33 0
61
题目大意:
给你一个长度为N的一个序列,然后让你找连续的包含m数的子序列共K个,让其和尽可能的大,求这个最大值。
思路:
1、考虑dp,设定dp【i】【j】表示现在dp到第j个数,其已经选好了i组数的最大值。
2、那么不难推出其状态转移方程:
①首先设定sum【i】表示以第i个数为止以从i-m到i作为一组数的和。
②dp【i】【j】=min(dp【i】【j】,dp【i-1】【k】+sum【i】);估计此时时间复杂度:O(n^2K),显然会超时,那么考虑进行优化:
我们上述递推状态转移方程式无需考虑dp到第j个位子的时候,到底要从哪个位子转移过来,是无脑枚举的位子k。
那么我们考虑从这个点进行优化。
③对应到达第j个位子的时候,如果我们有对这个数的两种选择:1,选择以j结尾的一组数纳入进去。2,不选择以j结尾的一组数。
那么此时可以将状态转移方程优化到只有两个:
dp【i】【j】=max(dp【i-1】【j-m】+sum【j】,dp【i】【j】)表示我们选择以j结尾的一组数纳入进去。
dp【i】【j】=max(dp【i】【j-1】)表示我们不选择以j结尾的一组数。
那么此时优化了一个数量级为常数,总时间复杂度:O(nk);
3、注意数据范围,需要使用long long int。
Ac代码:
#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
#define ll __int64
ll dp[5005][5005];
ll a[5005];
ll sum[5005];
int n,m,k;
int main()
{
while(~scanf("%d%d%d",&n,&m,&k))
{
for(int i=1;i<=n;i++)
{
scanf("%I64d",&a[i]);
if(i<=m)
{
sum[i]=sum[i-1]+a[i];
}
else
{
sum[i]=sum[i-1]+a[i]-a[i-m];
}
}
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
for(int kk=1;kk<=k;kk++)
{
if(i>=m)
{
dp[kk][i]=max(dp[kk][i-1],dp[kk][i]);
dp[kk][i]=max(dp[kk-1][i-m]+sum[i],dp[kk][i]);
}
}
}
printf("%I64d\n",dp[k][n]);
}
}