题目链接
题意
给定一个含有n个数字的序列,从中找到两段长度为k的不重叠的连续序列,使得这两段序列之和最大。
思路
先利用前缀和求出每段长度为k的大小用inte数组存储,然后对inte从后往前以长度为2*k开始遍历,第一个if用来更新区间为k的最大值,第二个if用来更新两个k区间的最大值。
代码
#include<iostream>
using namespace std;
#define ll long long
ll x[200005],anti[200005]={0},inte[200005];//x存输入的序列,anti存前缀和,inte为区间长度为k的大小
int main(){
ll n,k,i,mx=0,pa=0,pb=0,mxa=0,mxb=0;
cin>>n>>k;
for(i=1;i<=n;i++){
cin>>x[i];
anti[i]=anti[i-1]+x[i];
}
for(i=1;i<=n-k+1;i++){
inte[i]=anti[i+k-1]-anti[i-1];
}
ll preb;//不能少,用来存储当前区间为2k时最大的左端点
for(i=n-2*k+1;i>=1;i--){
if(mxb<=inte[i+k]){
mxb=inte[i+k];
pb=i+k;
}
if(mxa<=mxb+inte[i]){
mxa=mxb+inte[i];
pa=i;
preb=pb;
}
}
printf("%lld %lld",pa,preb);
return 0;
}