【题意】给定一个环形数组,求数组中连续长度不超过 K 的最大子段和。

【分析】

1. 题目中给定的是环形数组,其实可以在数组尾部复制一份原数据,于是把问题规模扩大到了 2 * N,但是方便了解题。

2. 对数据在加以转换:sum[i] 表示 1~i 数组元素的和,由于题目中给定了数据范围,保证了 int 不会溢出。

3. 单调队列里面的数据表示 x~x+k 范围内 sum[] 的最小值下标。于是 sum[x+k+1] - sum[deq[s]] 即是以 x+k+1 为结尾的子序列最大和。

【做法】要维护长度为k的一个单调递减的一个单调队列,O(1)的查询k,而不是枚举的O(n*k),复杂度优化为O(n)!

【总结】第一个单调队列优化的题目,还不是很熟悉!多做几个题吧!

【AC代码】


#include <stdio.h>
#include <string.h>
#include <iostream>
using namespace std;
const int unf = -0x3f3f3f3f;
const int maxn = 200010;
int a[maxn],que[maxn];//单调递增的单调队列
int sum[maxn];

int main()
{
    int n,k,T;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&k);
        for(int i=1; i<=n; i++) scanf("%d",&a[i]);
        for(int i=1; i<=n; i++){
            sum[i] = a[i];
            sum[i+n] = a[i];
        }
        for(int i=1; i<=2*n; i++){
            sum[i] = sum[i-1]+sum[i];
        }
        int st=0,ed=-1;
        int ans = unf;
        int sp,ep;
        for(int i=0; i<2*n; i++){
            //maintain queue!
            if(i>=k && sum[i-k]==sum[que[st]]) st++;
            while(st<=ed && sum[i]<sum[que[ed]]) ed--;
            que[++ed] = i;
            if(sum[i+1]-sum[que[st]]>ans){
                ans = sum[i+1] - sum[que[st]];
                sp = que[st]+1, ep = i+1;
            }
        }
        printf("%d %d %d\n",ans,(sp-1)%n+1,(ep-1)%n+1);
    }
    return 0;
}