最大连续和

【题目描述】
给你一个长度为 n 的整数序列{A1,A2…,An}
要求从中找出一段连续的长度不超过 m 的子序列,使得这个序列的和最大。

【输入格式】
第一行为两个整数 n,m;
第二行为n个用空格分开的整数序列,每个数的绝对值都小于1000.

【输出格式】
仅一个整数,表示连续长度不超过m的最大子序列和。

【输入样例】
6 4
1 -3 5 1 -2 3

【输出样例】
7

【数据范围】
对于 50% 的数据,1≤N,M≤10^4
对于 100% 的数据,1≤N,M≤2×10^5

朴素算法:
设F[i]为以第i个数为结尾的最大连续和,暴力枚举变量j
j=i-k+1→i 1≤k≤m,超时。

优化:
用空间换时间,把j的循环省掉,因为数据不是动态的,想到用前缀和。s[i]=a[1]+a[2]+…+a[i],但仍然超时。

再优化:
根据前缀和的思路
F[i]=max{s[i]-s[i-k]} ,其中1≤k≤m。
F[i]=s[i]-min{s[i-k]}
可以用RMQ维护s数组。
由于k是定长,可以用单调队列。
线段树或是堆均可。
此处代码为RMQ,其余方法请读者自行完成。

此题是数据结构优化动规的经典模板题。

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#define mem(a,b) memset(a,b,sizeof a)
using namespace std;

const int MAX_N=6e5+7;
const int Lg2=20;
const int Inf=0x3f;

int n,m;
int s[MAX_N],f[MAX_N];
int st[MAX_N][Lg2];
int ans=-0x3f3f3f3f;

void init()
{
	mem(s,0);
	mem(st,Inf);
}

int main()
{
	scanf("%d%d",&n,&m);
	init();
	s[0]=st[0][0]=0;
	for(int i=1;i<=n;i++)
	{
		int a;
		scanf("%d",&a);
		s[i]=s[i-1]+a;
		st[i][0]=s[i];
	}
	int level;
	for(level=1;(1<<level)<=n;level++)
	{
		for(int i=0;i<=n;i++)
		{
			st[i][level]=min(st[i][level-1],st[i+(1<<(level-1))][level-1]);
		}
	}
	for(int i=1;i<=n;i++)
	{
		int l=max(0,i-m);
		int r=i-1;
		int len=floor(1.0*log(r-l+1)/log(2));
		int m=min(st[l][len],st[r-(1<<len)+1][len]);
		f[i]=s[i]-m;
	}
	for(int i=1;i<=n;i++)
	{
		ans=max(ans,f[i]);
	}
	printf("%d\n",ans);
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值