POJ2828

一道線段樹的好題..poj2828
剛開始學線段樹,感覺進度有點落後了…
每個人都有一個value值和pos值,進入隊伍時會把pos上的人擠到後面去。
考慮逆序建立線段樹,因爲這樣插入之後可以保證位置不會再被改變。
插入時,如果左子節點已經站不下了,就站到右子節點。

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN  200010
#define lson rt<<1
#define rson rt<<1 | 1 
#define mid int m = (l + r) >> 1
int x[MAXN<<2];
int map[MAXN<<2],n;
int re[MAXN][2];
void build(int l,int r,int rt)
{
    if(l == r)
    {
        x[rt]  = 1;
        return;
    }
    int m = (l + r) >> 1;
    build(l,m,lson);
    build(m+1,r,rson);
    x[rt] = x[lson] + x[rson];
}
void update(int l,int r,int rt,int val,int pos)
{
    //cout<<l<<' '<<r<<' '<<rt<<' '<<val<<' '<<pos<<endl;    
    if(l == r)
    {
        map[rt] = val;
        x[rt] --;
        return;
    }
    mid;
    if(x[lson] > pos) update(l,m,lson,val,pos);
    else update(m+1,r,rson,val,pos-x[lson]);
    x[rt] = x[lson] + x[rson];
}
void print_ans(int l,int r,int rt)
{
    if(l == r)
    {
        if(l == n -1)
            cout<<map[rt]<<endl;
        else
            cout<<map[rt]<<' ';
        return;
    }
    mid;
    print_ans(l,m,lson);
    print_ans(m+1,r,rson);
}
int main()
{
    ios::sync_with_stdio(false);
    while(cin>>n){
        build(0,n-1,1);
        for(int i  = 0;i<n;i++){
            cin>>re[i][0]>>re[i][1];
        }
        for(int i = n - 1;i>-1;i--)
            update(0,n-1,1,re[i][1],re[i][0]);
        print_ans(0,n-1,1);
    }

}
發佈了35 篇原創文章 · 獲贊 10 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章