POJ 2828 Buy Tickets（神题！线段树or树状数组）

题目链接：POJ 2828 Buy Tickets

【题意】给了你 n（1<=n<=200000）个人的插队信息，让你输出最终的队列的排列

【思路】常规思考的话，这道题就是模拟，但是时间复杂度一定会很高。POJ的discuss上说这道题是神题，难得不是用什么数据结构，而是思路，这道题要逆向去思考，从最后一个人往前看，后插进来的人更容易定位，他一定能站到他想的位置，并且不会在挪动。再前一个人呢？他的位置即是接下来的空位的编号。于是有线段树用于维护位置信息。当然用树状数组也是可以做的，但是要加上二分来优化，也可以过。

下面是线段树代码：

/*
** POJ 2828 Buy Tickets
** Created by Rayn @@ 2014/05/08
** 神奇线段树，奇妙的逆向思维
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAX = 200005;

int pos[MAX], val[MAX], ans[MAX];
int tree[MAX*3], index;

void Build(int k, int l, int r)
{
    tree[k] = r - l + 1;    //开始时每个节点都有空位
    if(l == r)
        return ;
    int mid = (l + r) >> 1;
    Build(k<<1, l, mid);
    Build(k<<1|1, mid+1, r);
}
void Update(int k, int l, int r, int pos)
{
    tree[k]--;  //单点更新，减少一个空位
    if(l == r)
    {
        index = l;
        return ;
    }
    int mid = (l + r) >> 1;
    if(tree[k<<1] >= pos)   //如果当前位置的左边空位够的话就落在左边
    {
        Update(k<<1, l, mid, pos);
    }
    else
    {
        pos -= tree[k<<1];  //否则，就把左边的空位减去，再在右边定位
        Update(k<<1|1, mid+1, r, pos);
    }
}
int main()
{
#ifdef _Rayn
    freopen("in.txt", "r",stdin);
#endif
    int n;
    while(scanf("%d", &n) != EOF)
    {
        Build(1, 1, n);
        for(int i=1; i<=n; ++i)
        {
            scanf("%d%d", &pos[i], &val[i]);
        }
        for(int i=n; i>=1; --i)
        {
            Update(1, 1, n, pos[i]+1);
            ans[index] = val[i];
        }
        for(int i=1; i<=n; ++i)
        {
            printf("%d%c", ans[i], (i == n)? '\n':' ');
        }
    }
    return 0;
}

树状数组+二分的代码：

/*
** POJ 2828 Buy Tickets
** Created by Rayn @@ 2014/05/08
** 树状数组+二分
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAX = 200005;

int n, pos[MAX], val[MAX];
int tree[MAX], ans[MAX];

inline int Lowbit(int x)
{
    return x&(-x);
}
int GetSum(int idx)
{
    int sum = 0;
    while(idx > 0)
    {
        sum += tree[idx];
        idx -= Lowbit(idx);
    }
    return sum;
}
void Update(int idx, int val)
{
    while(idx <= n)
    {
        tree[idx] += val;
        idx += Lowbit(idx);
    }
}
int Search(int left, int right, int pos)
{
    while(left <= right)
    {
        int mid = (left + right) >> 1;
        int tmp = GetSum(mid);
        if(tmp == pos && ans[mid] == -1)
            return mid;
        if(tmp >= pos)
            right = mid - 1;
        else
            left = mid + 1;
    }
    return -1;
}
int main()
{
#ifdef _Rayn
    freopen("in.txt", "r",stdin);
#endif

    while(scanf("%d", &n) != EOF)
    {
        memset(tree, 0, sizeof(tree));
        memset(ans, -1, sizeof(ans));
        for(int i=1; i<=n; ++i)
        {
            scanf("%d%d", &pos[i], &val[i]);
            pos[i]++;
            Update(i, 1);    //所有位置填充1，代表有空位
        }
        for(int i=n; i>=1; --i)
        {
            int index = Search(pos[i], n, pos[i]);
            ans[index] = val[i];
            Update(index, -1);
        }
        for(int i=1; i<=n; ++i)
        {
            printf("%d%c", ans[i], (i == n)? '\n':' ');
        }
    }
    return 0;
}