动态规划,dp优化处理:人数计数问题

题面:

东东每个学期都会去寝室接受扫楼的任务,并清点每个寝室的人数。每个寝室里面有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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值