//
// 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;
}
poj 3667 线段树-区间的前缀和后缀
最新推荐文章于 2022-04-06 16:25:15 发布