E44 单调队列优化DP 修剪草坪

视频链接：452 单调队列优化DP 修剪草坪_哔哩哔哩_bilibili

LOJ10177 修剪草坪

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long LL;
const int N=1e5+10;
int n,k,q[N];
LL w[N],f[N],sum;

int main(){
  cin>>n>>k; k++; //
  for(int i=1;i<=n;i++) cin>>w[i],sum+=w[i];
  
  int h=1,t=0; LL s=1e18;
  for(int i=1;i<=n;i++){
    while(h<=t && f[q[t]]>=f[i-1]) t--;
    q[++t]=i-1;
    if(q[h]<i-k) h++;
    f[i]=f[q[h]]+w[i];
    if(i>n-k) s=min(s,f[i]);
  }
  cout<<sum-s;
}

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long LL;
const int N=1e5+10;
int n,k,q[N];
LL w[N],f[N],sum;

int main(){
  cin>>n>>k; k++; //
  for(int i=1;i<=n;i++) cin>>w[i],sum+=w[i];
  
  int h=1,t=0; LL s=1e18;
  for(int i=1;i<=n;i++){
    while(h<=t && q[h]<i-k) h++;
    while(h<=t && f[q[t]]>=f[i-1]) t--;
    q[++t]=i-1;
    f[i]=f[q[h]]+w[i];
    if(i>n-k) s=min(s,f[i]);
  }
  cout<<sum-s;
}