题目描述
给你一个长度为 n 的整数序列 {A1,A2,⋯,An},要求从中找出一段连续的长度不超过 m 的非空子序列,使得这个序列的和最大。
输入格式
第一行为两个整数 n,m;
第二行为 n 个用空格分开的整数序列,每个数的绝对值都小于 1000。
输出格式
仅一个整数,表示连续长度不超过 m 的最大非空子序列和。
输入样例
6 4
1 -3 5 1 -2 3
输出样例
7
参考思路
本题我采用单调队列的方法,如果不了解单调队列的话可以先去CSDN搜索一下。
我们可以先求出数组的前缀和,然后建立一个单调递增队列。那么队列的头肯定是前缀和中最小的那一个,那么我们求值的时候只需要遍历n,让当前的前缀和减去队列的头(既然你是前缀和,你这个前缀和队列的头最小,你求区间和的时候肯定减去这个最小的值),每一次都记录得到最大子序列和ans,,这样可以形成一个区间。
我们再思考一下尾出队和头出队的方式,那么这题便可解决。
参考代码
#define _CRT_SECURE_NO_WARNINGS//vs硬伤
#include<cstdio>
#include<cmath>
#include<climits>//最值头文件
#include<iostream>
using namespace std;
int n, m;
int s[300090];//前缀和数组
int num[300090];//队列数组
int main() {
int x;
int head = 1, tail = 1;
int ans = LONG_MIN;//答案初始为一个最小值
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &x);
s[i] = s[i - 1] + x;//得到前缀和
}
for (int i = 1; i <= n; i++) //队列采用递增方式,然后用前缀和s[i]-队列中的头(即前缀和最小值)
{
while (head <= tail && num[head] < i - m) //判断头是否出队 这里要特别注意一下区间是不大于m
head++;
while (head <= tail && s[num[tail]] >= s[i])//判断尾是否出队 如果即将入队的那个元素比尾小,那么尾就要出队
tail--;
ans = max(ans, s[i] - s[num[head]]);//得到解
num[++tail] = i;//入队
}
printf("%d", ans);
return 0;
}
如果有更好的方法或者有错误的地方,欢迎斧正。
感谢您的阅读!(*^▽^*)