[線段樹] POJ 2828 - Buy Tickets

題意:

輸入一個整數n。接下來n行分別輸入兩個整數Posi , Vali.。分別代表要插入的位置 和 插入的值。

思路:

本題比較難想,暴力會超時,倒着插入會避免移動操作,只需要用線段樹查詢空着的個數即可。

#include <stdio.h>
#include <iostream>
using namespace std;
#define lson rt << 1
#define rson rt << 1 | 1
int maxn[200000 << 2], ans[200005];
void build(int l, int r, int rt)
{
    if(l == r)
    {
        maxn[rt] = 1;
        return ;
    }
    int mid = (l + r) >> 1;
    build(l, mid, lson);
    build(mid + 1, r, rson);
    maxn[rt] = maxn[lson] + maxn[rson];
}
void query(int l, int r, int rt, int ret, int val)
{
    if(l == r)
    {
        ans[l] = val;
        maxn[rt] = 0;
        return ;
    }
    int mid = (l + r) >> 1;
    if(ret <= maxn[lson])
    {
        query(l, mid, lson, ret, val);
    }
    else
    {
        query(mid + 1, r, rson, ret - maxn[lson], val);
    }
    maxn[rt] = maxn[lson] + maxn[rson];
}
int main()
{
    int n;
    pair<int, int> p[200000];
    while(~scanf("%d", &n))
    {
        for(int i = 0; i < n; i++)
        {
            scanf("%d %d", &p[i].first, &p[i].second);
        }
        build(1, n, 1);
        for(int i = n - 1; i >= 0; i--)
        {
            query(1, n, 1, p[i].first + 1, p[i].second);
        }
        for(int i = 1; i <= n; i++)
        {
            printf(i == n? "%d " : "%d\n", ans[i]);
        }
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章