题面:
东东每个学期都会去寝室接受扫楼的任务,并清点每个寝室的人数。每个寝室里面有ai个人(1<=i<=n)。从第i到第j个宿舍一共有sum(i,j)=a[i]+…+a[j]个人。东东需要扫楼m次,每一次数第i到第j个宿舍sum(i,j)
问题是要找到sum(i1, j1) + … + sum(im,jm)的最大值。且ix <= iy <=jx和ix <= jy <=jx的情况是不被允许的。也就是说m段都不能相交。
注:1 ≤ i ≤ n ≤ 1e6 , -32768 ≤ ai ≤ 32767 人数可以为负数(1<=n<=1000000)
输入m,输入n。后面跟着输入n个ai
输出最大和
sample input:
1 3 1 2 3
2 6 -1 4 -2 3 -2 3
sample output:
6
8
注意:
数据量很大,需要scanf读入和dp处理
思路:
利用动态规划的思想,因为要求m个子段都不能相交,所以可以利用状态转移方程dp[i][j] = max(dp[i][j-1]+a[j] , dp[i-1][k]+a[j])来计算,dp[i][j]必须取到第j个元素,否则这种情况就肯定会在其他情况中被枚举过。前一个状态表示将a[j]加入dp子串,后一种表示单独将a[j]作为一个子串
这样做的话需要三个循环,由于本题的数据量很大,这样做会爆掉。
观察转移方程发现,其实每次状态转移后的结果是上一次比较中较大的值加上a[j],所以可以创建一个数组premax对于从 j 个元素中寻找 i-1 段中的最大值,dp记录从 j 个元素中找 i 段的最大子串和
同时可以将dp降为一维滚动数组形式,这样的话转移方程就变成了dp[j]=max(dp[j-1]+a[j],premax[j-1]+a[j])
注意:每次完成计算后记得更新premax来为下一次计算做准备,每一轮开始计算一个新子串的时候初始化当前的ans。
#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1e6+1;
const int INF=1e9+1;
int ans=-INF;
int a[N];
int dp[N];
int premax[N];
int main()
{
int m,n;
while(scanf("%d%d",&m,&n)!=EOF)
{
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
memset(premax,0,sizeof(premax));
for(int i=0;i<m;i++)
{
ans=-INF;
for(int j=i;j<n;j++)
{
dp[j]=max(dp[j-1]+a[j],premax[j-1]+a[j]);
premax[j-1]=ans;
ans=max(ans,dp[j]);
}
}
cout<<ans<<endl;
}
return 0;
}