POJ 2828 Buy Tickets

题目

Railway tickets were difficult to buy around the Lunar New Year in China, so we must get up early and join a long queue…

The Lunar New Year was approaching, but unluckily the Little Cat still had schedules going here and there. Now, he had to travel by train to Mianyang, Sichuan Province for the winter camp selection of the national team of Olympiad in Informatics.

It was one o’clock a.m. and dark outside. Chill wind from the northwest did not scare off the people in the queue. The cold night gave the Little Cat a shiver. Why not find a problem to think about? That was none the less better than freezing to death!

People kept jumping the queue. Since it was too dark around, such moves would not be discovered even by the people adjacent to the queue-jumpers. “If every person in the queue is assigned an integral value and all the information about those who have jumped the queue and where they stand after queue-jumping is given, can I find out the final order of people in the queue?” Thought the Little Cat.

输入

There will be several test cases in the input. Each test case consists of N + 1 lines where N (1 ≤ N ≤ 200,000) is given in the first line of the test case. The next N lines contain the pairs of values Posi and Vali in the increasing order of i (1 ≤ i ≤ N). For each i, the ranges and meanings of Posi and Vali are as follows:

• Posi ∈ [0, i − 1] — The i-th person came to the queue and stood right behind the Posi-th person in the queue. The booking office was considered the 0th person and the person at the front of the queue was considered the first person in the queue.
• Vali ∈ [0, 32767] — The i-th person was assigned the value Vali.

There no blank lines between test cases. Proceed to the end of input.

输出

For each test cases, output a single line of space-separated integers which are the values of people in the order they stand in the queue.

思路

利用树状数组维护每个点之前剩余的空位，再从后往前遍历找到合适位置

代码

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

const int MAXN = 200005;
int line[MAXN];
int N;

int lowbit(int x)
{
	return x&(-x);
}

void add(int index, int addnum)
{
	while(index <= MAXN)
	{
		line[index] += addnum;
		index += lowbit(index);
	}
}

void showtree()
{
	for(int i = 1; i <= N; i++) printf("%d ", i);
	printf("\n");
}

int sum(int index)
{
	int ans = 0;
	while(index > 0)
	{
		ans += line[index];
		index -= lowbit(index);
	}
	
	return ans;
}

int find(int pre)
{
	int l = 1;
	int r = N;
	int mid = (l + r) / 2;
	
	while(l < r)
	{
		mid = (l + r) / 2;
		if(sum(mid) < pre) l = mid + 1;
		else  r = mid;		
	}
	
	return l;
}

int pos[MAXN], val[MAXN], order[MAXN];

int main()
{
	while(scanf("%d", &N) != EOF)
	{
		memset(line, 0, sizeof(line));
		for(int i = 1; i <= N; i++) add(i, 1);
		//showtree();
		for(int i = 1; i <= N; i++)
		{
			scanf("%d %d", &pos[i], &val[i]);
		}
		
		for(int i = N; i >= 1; i--)
		{
			int loc = find(pos[i]+1);
			order[loc] = val[i];
			add(loc, -1);
			//printf("%d is at %d\n", val[i], loc);
			//showtree();
		}
		
		for(int i = 1; i <= N; i++) printf("%d ", order[i]);
		printf("\n");
	}
	
	
	
	return 0;
}