SDU 程序设计思维与实践 Week13 最大连续子段和问题【单调队列优化DP】

最大连续子段和问题

TT 猫咖的生意越来越红火，人越来越多，也越来越拥挤。
为了解决这个问题，TT 决定扩大营业规模，但猫从哪里来呢？
TT 第一时间想到了神秘人，想要再次通过完成任务的方式获得猫咪。
而这一次，神秘人决定加大难度。
给定一个环，A[1], A[2], A[3], … , A[n]，其中 A[1] 的左边是 A[n]。要求从环上找出一段长度不超过 K 的连续序列，使其和最大。
这一次，TT 陷入了沉思，他需要你们的帮助。

输入

第一行一个整数 T，表示数据组数，不超过 100。
每组数据第一行给定两个整数 N K。（1 ≤ N ≤ 100000, 1 ≤ K ≤ N）
接下来一行，给出 N 个整数。（-1000 ≤ A[i] ≤ 1000）。

输出

对于每一组数据，输出满足条件的最大连续和以及起始位置和终止位置。
如果有多个结果，输出起始位置最小的，如果还是有多组结果，输出长度最短的。

输入样例

4
6 3
6 -1 2 -6 5 -5
6 4
6 -1 2 -6 5 -5
6 3
-1 2 -6 5 -5 6
6 6
-1 -1 -1 -1 -1 -1

输出样例

7 1 3
7 1 3
7 6 2
-1 1 1

思路分析

• 设$f[i]$表示以$i$为结尾的最大连续子段和
• 答案为 m a x { f [ i ] } max\{f[i]\} max{f[i]}
• 转移方程可列为： f [ i ] = s u m [ i ] − m i n { s u m [ k ] } ( i − m ≤ k ≤ i ) f[i]=sum[i]-min\{sum[k]\}(i-m\leq k \leq i) f[i]=sum[i]−min{sum[k]}(i−m≤k≤i)
• 利用单调队列处理 m i n { s u m [ k ] } min\{sum[k]\} min{sum[k]}
• 需要特殊处理$i==q.front()$，解决的方法是先维护窗口大小，再维护单调性

代码

#include<iostream>
#include<cstdio>
#include<deque>
#include<cstring>
using namespace std;
const int maxsize= 200010,inf=1e8;
int t,n,k,l,r;
int a[maxsize];
long long sum[maxsize],ans;
deque<int>q;
int main(){
	cin>>t;
	while(t--){
		cin>>n>>k;
		memset(a,0,sizeof a);
		memset(sum,0,sizeof sum);
		ans=-inf;
		l=r=0;
		q.clear();
		q.push_back(0);
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			a[i+n]=a[i];
		}
		for(int i=1;i<=2*n-1;i++){
			sum[i]=sum[i-1]+a[i];
		}
		for(int i=1;i<=2*n-1;i++){
			while(q.size()&&q.front()<i-k) q.pop_front();
			long long temp=sum[i]-sum[q.front()];
			if(temp>ans){
				ans=temp;
				l=q.front()+1;
				r=i;
				if(l>n) l-=n;
				if(r>n) r-=n;
			}
			while(q.size()&&sum[q.back()]>sum[i]) q.pop_back();
			q.push_back(i);
		}
		cout<<ans<<" "<<l<<" "<<r<<endl;
	}	 
}