POJ 2828 Buy Tickets

本题使用了一个巧妙的构造方法,首先,如果按顺序插队的话,序列是无法确定的,因为插队的时候被插的人的位置会改变。所以可以倒过来想,最后一个人的位置必然可以确定,然后接着是倒数第二的,依次的位置就都出来了。

然后线段树的功能是统计区间内可插的位置数吧,我们可以先想想叶子节点,如果这个位置被确定了,那么该叶子节点的值应该变为0了,就相当于有了一个新的队列,是不包含值为0的叶子节点的队列,那么再插的时候,不管这些为0的叶子节点,在值为1的中寻找第i个位置,插进去,就又确定了一个人的位置。


/*
ID: sdj22251
PROG: subset
LANG: C++
*/
#include <iostream>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <cmath>
#include <ctime>
#define LOCA
#define MAXN 200005
#define INF 100000000
#define eps 1e-7
#define L(x) x<<1
#define R(x) x<<1|1
using namespace std;
struct node
{
    int left, right, mid, cnt;
}tree[4 * MAXN];
int a[MAXN], b[MAXN];
int ans[MAXN];
void make_tree(int s, int e, int C)
{
    tree[C].left = s;
    tree[C].right = e;
    tree[C].mid = ((s + e) >> 1);
    tree[C].cnt = tree[C].right - tree[C].left + 1;
    if(s == e) return;
    make_tree(s, tree[C].mid, L(C));
    make_tree(tree[C].mid + 1, e, R(C));
}
void insert(int p, int v, int C)
{
    tree[C].cnt--; //进入区间后区间值要减1
    if(tree[C].right == tree[C].left)
    {
        ans[tree[C].right] = v;
        return;
    }
    if(tree[L(C)].cnt >= p)
    insert(p, v, L(C));
    else insert(p - tree[L(C)].cnt, v, R(C));  //左侧不够,再算上右侧的
}
int main()
{
#ifdef LOCAL
    freopen("d:/data.in","r",stdin);
    freopen("d:/data.out","w",stdout);
#endif
    int n, i;
    while(scanf("%d", &n) != EOF)
    {
        make_tree(1, n, 1);
        for(i = 0; i < n; i++)
        {
            scanf("%d%d", &a[i], &b[i]);
        }
        for(i = n - 1; i >= 0; i--)
        {
            insert(a[i] + 1, b[i], 1);
        }
        for(i = 1; i <= n; i++)
        printf("%d ", ans[i]);
        printf("\n");
    }
    return 0;
}


你可能感兴趣的:(c,tree,insert)