【codeforces】338E. Optimize! 【线段树】

传送门：【codeforces】338E. Optimize!

题目分析：
这题有两种解法，一个$O(n\sqrt n)$的，一个$O(nlogn)$。
首先我们要转换下题意，题目描述的函数，我们可以这么考虑，a数组区间[L,L+len-1]中最大的元素和b数组中最小的元素，a中第二大的元素和b中第二小的元素……每一对的和都大于等于h。我们让b[i] = h - b[i]，然后就是a中区间[L,L+len-1]内所有第k大的元素都大于等于b中第k大的元素。

1.$O(n\sqrt n)$
首先将b数组按照上面处理后使得b[i] = h - b[i]，然后对b数组升序排序。另b数组对应val数组，val数组初始化元素均为-1，val[i]=-1表示第i个位置没有a中元素比其大，val[i]>=0表示第i歌位置，a中存在val[i]+1个怨毒比其大。
然后对要插入的a[i]寻找最大的小于等于a[i]的b中元素的位置x，将这个位置的val[x]++，表示这个a[i]可以被选择和1~x中的其中一个b中元素匹配。插入完以后我们对求后缀和，如果每个位置的后缀和都大于等于0，那么匹配大小就是len，可以一一对应，那么本次就是合法的，答案加一，否则本次不合法。对于后缀和的快速求，我们需要维护一个块的和，以及这个块内部后缀和的最小值。然后就可以$O(\sqrt n)$的复杂度求出是不是每个后缀和都大于等于0。对于更新，就暴力重构所在的块就行了。

2.$O(nlogn)$
其实，这个就是把上面的改了一下，换成线段树维护后缀和了，然后维护区间最小值即可。

$my~~code:$

#include <bits/stdc++.h>
using namespace std ;

typedef long long LL ;

#define clr( a , x ) memset ( a , x , sizeof a )
#define ls ( o << 1 )
#define rs ( o << 1 | 1 )
#define lson ls , l , m
#define rson rs , m + 1 , r
#define mid ( ( l + r ) >> 1 )
#define root 1 , 1 , m

const int MAXN = 150005 ;

int n , m , h ;
int f[MAXN << 2] , minv[MAXN << 2] ;
int a[MAXN] , b[MAXN] ;

int search ( int x , int l = 1 , int r = m ) {
    while ( l < r ) {
        int t = ( l + r + 1 ) >> 1 ;
        if ( b[t] <= x ) l = t ;
        else r = t - 1 ;
    }
    return l ;
}

void build ( int o , int l , int r ) {
    f[o] = 0 ;
    minv[o] = l - m - 1 ;
    if ( l == r ) return ;
    int m = mid ;
    build ( lson ) ;
    build ( rson ) ;
}

void pushdown ( int o ) {
    if ( f[o] ) {
        f[ls] += f[o] ;
        f[rs] += f[o] ;
        minv[ls] += f[o] ;
        minv[rs] += f[o] ;
        f[o] = 0 ;
    }
}

void update ( int L , int R , int v , int o , int l , int r ) {
    if ( L <= l && r <= R ) {
        minv[o] += v ;
        f[o] += v ;
        return ;
    }
    pushdown ( o ) ;
    int m = mid ;
    if ( L <= m ) update ( L , R , v , lson ) ;
    if ( m <  R ) update ( L , R , v , rson ) ;
    minv[o] = min ( minv[ls] , minv[rs] ) ;
}


void solve () {
    int ans = 0 ;
    for ( int i = 1 ; i <= m ; ++ i ) {
        scanf ( "%d" , &b[i] ) ;
        b[i] = h - b[i] ;
    }
    sort ( b + 1 , b + m + 1 ) ;
    build ( root ) ;
    for ( int i = 1 ; i <= n ; ++ i ) {
        scanf ( "%d" , &a[i] ) ;
        if ( i > m && a[i - m] >= b[1] ) update ( 1 , search ( a[i - m] ) , -1 , root ) ;
        if ( a[i] >= b[1] ) update ( 1 , search ( a[i] ) , 1 , root ) ;
        if ( i >= m && minv[1] >= 0 ) ++ ans ;
    }
    printf ( "%d\n" , ans ) ;
}

int main () {
    while ( ~scanf ( "%d%d%d" , &n , &m , &h ) ) solve () ;
    return 0 ;
}