poj 3667 线段树-区间的前缀和后缀

//
//  main.cpp
//  poj 3667 线段树-区间的前缀和后缀
//
//  Created by XD on 15/9/13.
//  Copyright (c) 2015年 XD. All rights reserved.
//
/*
 线段树的区间更新
 这里需要注意的是线段树的每个节点应该保存什么样的属性。
 这里是要找到一个区间使得其长度为d，且左边的值最小。我们保持线段树的节点的sum属性，pre属性，和suffix属性。
 这里的pre，和suffix的更新是关键的地方，得注意当前节点的pre和suffix可以用其左右儿子节点来更新。
 */
#include <iostream>
#include <queue>
#include <stack>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#include <cstdio>
#define ll long long
using namespace std ;
const int maxn = 50000 * 4 + 10 ;
int d ;
struct Intervaltree
{
    int _sum[maxn] , _pre[maxn] , _suffix[maxn],_set[maxn];
    int  qr,ql ,v,op,ans ;
    void build(int o , int L , int R)
    {
        if(L <  R)
        {
            _pre[o] = _suffix[o] = R - L +1  ;
            int m = L + (R-L)/2 ;
            build(o*2, L  , m)  ;
            build(o*2+1, m+1,R ) ;
        }
        else{
            _pre[o] = _suffix[o] =1 ;
        }
    }
    void init()
    {
        memset(_set, -1, sizeof(_set)) ;
        memset(_sum, 0, sizeof(_sum)) ;
    }
    void maintain(int o,int L , int R)
    {
        int lc = o * 2 ; int rc = o *2+1 ;
        int m =   L + (R - L ) /2 ;
        if(L < R){
            _sum[o] = _sum[lc] + _sum[rc] ;
            _pre[o] = _pre[lc] == m -  L + 1  ? _pre[lc] + _pre[rc]:_pre[lc] ;
            _suffix[o] = _suffix[rc] == R-m ? _suffix[lc] + _pre[rc]:_suffix[rc] ;
        }
        if(_set[o]==0 )
        {
            _sum[o] = 0 ;
            _pre[o] = _suffix[o] = R- L + 1 ;
        }
        else if (_set[o] >0)
        {
            _sum[o] = R - L +1 ;
            _pre[o] = _suffix[o] = 0 ;
        }
    }
    void pushdown(int o )
    {
        if(_set[o] >= 0 )
        {
            _set[o*2] = _set[o*2 +1] = _set[o] ;
            _set[o] = -1 ;
            
        }
    }
    void update(int o , int L , int R)
    {
        int lc = o *2 ; int rc = o *2+1 ;
        if(ql <= L && R <= qr)
        {
            _set[o] = v ;
        }
        else{
            pushdown(o) ;
            int m =  L + (R - L ) /2 ;
            if(ql <= m ) update(lc, L, m) ; else maintain(lc, L  , m) ;
            if(qr>m) update(rc, m+1, R)  ;else maintain(rc, m+1, R) ;
        }
        maintain(o , L , R) ;
    }
    void findLeft(int o , int L , int R)
    {
        if(R - L + 1< d || R - L + 1 - _sum[o] < d)
        {
            return ;
        }
        if(_set[o] == 0 )
        {
            ans = L ;
        }
        else if (_set[o] ==1 )
        {
            return ;
        }
        else{
            int m  = L + (R -L) /2 ;
            findLeft(o *2 ,L, m) ;
            if(ans > 0 )
            {
                return ;
            }
            if(_suffix[o*2] +_pre[o*2+1] >= d)
            {
                if(_suffix[o*2] > 0 ) ans =  m - _suffix[o*2] +1 ;
                else ans = m +1 ;
                return ;
            }
            findLeft(o*2 +1 , m + 1, R) ;
        }
    }
};
Intervaltree tree ;
int main() {
    int n , m ,cmd;
    scanf("%d%d" ,&n,&m) ;
    tree.init() ;
    tree.build(1, 1, n) ;
    for (int i = 0; i <  m; i++) {
        scanf("%d" ,&cmd) ;
        if (cmd == 1) {
            scanf("%d" , &d) ;
            tree.ans = 0 ;
            tree.findLeft(1, 1, n) ;
            printf("%d\n" ,tree.ans) ;
            if (tree.ans > 0 ) {
                tree.v = 1 ;
                tree.ql = tree.ans ; tree.qr = tree.ans + d -1 ;
                tree.update(1, 1, n) ;
            }
        }
        else{
            scanf("%d%d" ,&tree.ans ,&d) ;
            tree.v = 0 ;
            tree.ql = tree.ans ; tree.qr = tree.ql + d - 1 ;
            tree.update(1, 1, n) ;
        }
    }
       return 0;
}