第一次写题解呢,这道题一开始没用dp做,但是写的代码一直wa,看了别人代码自己写了下
用到dp的思想,枚举第二个子列,然后每次比较当前的pre(第二个子列和这个子列前最大的第一个子列之和),得到更大的便更新pre
代码如下:
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxn 200005
long long sum[maxn];
int map[maxn];
int main(void)
{
long long n,k,p1,p2,s,e,st;
cin >> n >> k;
for(int i = 1; i <= n;i++)
{
cin >> map[i];
sum[i] = sum[i-1] + map[i];
}
long long pre1 = -1,pre = -1;
for(int i = k + 1; i <= n-k+1;i++)
{
p1=sum[i-1]-sum[i-k-1];
p2=sum[i+k-1]-sum[i-1];
if(p1 > pre1)
pre1 = p1,s = i-k;//如果得到更大的第一个子序列就更新
if(pre1+p2>pre)
{
st = s;//更新当前pre最大时,更新第一个子序列的位置
e = i;
pre = pre1 + p2;
}
}
cout << st << ' ' << e << endl;
}