poj 2828【线段树 单点更新】

POJ 2828

还是弱啊。思维是个好东西。。。

刚开始想来想去用线段树存人的话不仅超时，而且存不下。。。居然是存空位！

sum[]数组存这个序列空位个数，然后逆序遍历。逆序好理解，毕竟最后一个人插进来位置已经可以确定了，前面的位置就根据他来更新。

#include<iostream>
#include<cstdio>
#define mid int m=(l+r)>>1
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int maxn=200005;
int sum[maxn<<2],ans[maxn<<2];
int N,p[maxn],v[maxn];


void Build(int l,int r,int rt)//建树存空位
{
    sum[rt]=r-l+1;//初始建树时线段对应的空位数就是线段长度(点看成长度为1的线段)
    if(l==r) return;
    mid;
    Build(lson);
    Build(rson);
}

void PushUp(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void Update(int pos,int val,int l,int r,int rt)
{
    if(l==r){
        ans[l]=val;//递归到节点，用ans[]数组记录第l个人的val值。
        sum[rt]--;//空位数减1
        return;
    }
    mid;
    if(pos<=sum[rt<<1]) Update(pos,val,lson);//如果当前节点左孩子的空位数足够，就递归左孩子
    else Update(pos-sum[rt<<1],val,rson);//如果左孩子的空位数不够，那么就递归又孩子，但空位数要变成pos-sum[rt<<1]
    PushUp(rt);//向上更新父节点的空位数
}

int main()
{
    while(scanf("%d",&N)==1)
    {
        Build(1,N,1);
        for (int i=1;i<=N;i++) scanf("%d%d",&p[i],&v[i]);
        for (int i=N;i>=1;i--){
            Update(p[i]+1,v[i],1,N,1);//逆序遍历 
        }
        for (int i=1;i<N;i++) printf("%d ",ans[i]);
        printf("%d
",ans[N]);
    }
    return 0;
}