CF Manthan, Codefest 19E. Let Them Slide // 线段树极值+差分

题目

题意

给这样一个东西，有n行，然后每行不一定等长，每行可以随意水平滑动，求每一列的最大值（每一列的最大值都独立）。

思路

可以做一个等效操作
考虑每一行，$[len[i]+1,w-len[i]-1]$区间里一定可以得到数列里的最大值。
那么可以单独处理每一列的$[1,len[i]]$和$[max(len[i]+1,w-len[i]+1),w]$的极值（线段树），中间的部分用差分，打上数列最大值的标记。
这里需要注意，统计答案时，每一行可以不选数列里的值，那么在数列左右添加一个零。
这样复杂度就可以降为$O(2w\ast logw+w)$

/*   Author : Rshs
 *   Data : 2019-09-26-11.17
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const double pai = acos(-1);
const double eps = 1e-10;
const LL mod = 1e9+7;
const int MN = 1e6+5;
int c[MN];
vector<int>v[MN];
LL cf[MN];int n,w;
/****************************************************************************/
int T[MN<<2];
inline void pushUp(int rt){
    T[rt]=max(T[rt<<1],T[rt<<1|1]);
}
void build(int l,int r,int rt,int h){
    if(l==r){
        T[rt]=v[h][l];
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1,h);
    build(m+1,r,rt<<1|1,h);
    pushUp(rt);
}
int query(int L,int R,int l,int r,int rt){
    if(l>=L&&r<=R)
        return T[rt];
    int m=(l+r)>>1;
    int re=INT_MIN;
    if(m>=L) re=max(re,query(L,R,l,m,rt<<1));
    if(m<R)  re=max(re,query(L,R,m+1,r,rt<<1|1));
    return re;
}
/****************************************************************************/
void deal(int h){
    build(0,c[h]+1,1,h);
    for(int i=1;i<=c[h];i++){
        int r=i;
        int l=max(0,i-(w-c[h]));
        LL gg=query(l,r,0,c[h]+1,1);
        cf[i]+=gg;cf[i+1]-=gg;
    }
    for(int i=max(c[h]+1,w-c[h]+1);i<=w;i++){
        int l=c[h]-(w-i);
        int r=min(i,c[h]+1);
        LL gg=query(l,r,0,c[h]+1,1);
        cf[i]+=gg;cf[i+1]-=gg;
    }
    if(c[h]+1<=(w-c[h]+1)-1){
        LL gg=query(0,c[h]+1,0,c[h]+1,1);
        cf[c[h]+1]+=gg;
        cf[w-c[h]+1]-=gg;
    }
}
int main(){
    cin>>n>>w;
    for(int i=1;i<=n;i++){
        scanf("%d",&c[i]);
        v[i].push_back(0);
        for(int j=1;j<=c[i];j++) {
            int sa;scanf("%d",&sa);v[i].push_back(sa);
        }
        v[i].push_back(0);
    }
    for(int i=1;i<=n;i++){
        deal(i);
    }
    for(int i=1;i<=w;i++) cf[i]=cf[i]+cf[i-1],cout<<cf[i]<< ' ';
    return 0;
}