Codeforces 1304F2 Animal Observation (hard version) 代码（dp滑动窗口线段树区间更新优化）

https://codeforces.com/contest/1304/problem/F2

 #include<bits/stdc++.h>
 using namespace std;
 const int maxn = 4e4+;
 int dp[][maxn];
 int val[][maxn];
 int sum[][maxn];
 int n,m,k;
 struct node{
     int l,r;
     int Max,lz;
 }seg_t[maxn*];
 void build(int l,int r,int p){
     seg_t[p].l = l,seg_t[p].r = r;
     if(l == r) {
         seg_t[p].Max = ;return ;
     }
     int mid = (l+r)>>;
     build(l,mid,p*);
     build(mid+,r,p*+);
     seg_t[p].Max = max(seg_t[p*].Max ,seg_t[p*+].Max );
 }
 void pushdown(int k){
     seg_t[k*].lz +=seg_t[k].lz ;
     seg_t[k*+].lz +=seg_t[k].lz ;
     seg_t[k*].Max +=seg_t[k].lz ;
     seg_t[k*+].Max +=seg_t[k].lz ;
     seg_t[k].lz = ;
 }
 void upd(int L,int R,int p,int v){
     if(seg_t[p].l == L && seg_t[p].r == R){
         seg_t[p].lz +=v;
         seg_t[p].Max +=v;
         return;
     }
     if(seg_t[p].lz ) pushdown(p);
     int mid = (seg_t[p].l + seg_t[p].r )>>;
     if(R<=mid) upd(L,R,p*,v);
     else if(L>mid) upd(L,R,p*+,v);
     else{
         upd(L,mid,p*,v);
         upd(mid+,R,p*+,v);
     }
     seg_t[p].Max = max(seg_t[p*].Max ,seg_t[p*+].Max );
 }
 int query(int p,int L,int R){
     if(seg_t[p].l == L && seg_t[p].r == R) return seg_t[p].Max ;
     if(seg_t[p].lz ) pushdown(p);
     int mid = (seg_t[p].l +seg_t[p].r)>>;
     if(R<=mid) return query(p*,L,R);
     else if(L>mid) return query(p*+,L,R);
     else return max(query(p*,L,mid),query(p*+,mid+,R));
 }
 int main(){
     cin>>n>>m>>k;
     for(int i = ;i<=n;i++){
         for(int j = ;j<=m;j++) {
             cin>>val[i][j];
             sum[i][j] = sum[i][j-] + val[i][j];
         }
     }
     for(int i = ;i<=m-k+;i++){
         dp[][i] = sum[][i+k-] - sum[][i-]+sum[][i+k-]-sum[][i-];
     }
     for(int i = ;i<=n;i++){
         memset(seg_t,,sizeof(seg_t));
         build(,*m,);
         for(int j = ;j<=m;j++) upd(j,j,,dp[i-][j]);
         for(int j = ;j<=k;j++) upd(,j,,-val[i][j]);
         for(int j = ;j<=m-k+;j++){
             dp[i][j] = max(dp[i][j],query(,,m)+sum[i][j+k-]-sum[i][j-]+sum[i+][j+k-]-sum[i+][j-]);
             upd(max(,j-k+),j,,val[i][j]);
             upd(j+,j+k,,-val[i][j+k]);
         }
     }
     int ans = ;
     for(int i = ;i<=m;i++) ans = max(ans,dp[n][i]);
     cout<<ans;
     return ;
 }