Cod sursa(job #641315)

Utilizator yamahaFMI Maria Stoica yamaha Data 27 noiembrie 2011 20:53:10
Problema Heapuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.28 kb
#include <stdio.h>
#include <assert.h>

#define maxn 200005

int A[maxn], Heap[maxn], Pos[maxn], l, n, nr;

void push(int x)
{
	int aux;
	while (x/2 && A[Heap[x]]<A[Heap[x/2]])
	{
		aux = Heap[x];
		Heap[x] = Heap[x/2];
		Heap[x/2] = aux;

		Pos[Heap[x]] = x;
		Pos[Heap[x/2]] = x/2;
		x /= 2;
	}
}

void pop(int x)
{
	int aux, y = 0;
	while (x != y)
	{
		y = x;
		if (y*2<=l && A[Heap[x]]>A[Heap[y*2]]) x = y*2;
		if (y*2+1<=l && A[Heap[x]]>A[Heap[y*2+1]]) x = y*2+1;
		aux = Heap[x];
		Heap[x] = Heap[y];
		Heap[y] = aux;
		Pos[Heap[x]] = x;
		Pos[Heap[y]] = y;
	}
}

int main()
{
	freopen("heapuri.in", "r", stdin);
	freopen("heapuri.out", "w", stdout);

	int i, x, cod;
	scanf("%d ", &n);
	assert(1<=n && n<=200000);
	for (i=1; i<=n; i++)
	{
		scanf("%d ", &cod);
		assert(1<=cod && cod<=3);
		if (cod < 3) 
		{
			scanf("%d ", &x);
			assert(1<=x && x<=1000000000);
		}

		if (cod == 1)
		{
			l++, nr++;
			A[nr] = x;
			Heap[l] = nr;
			Pos[nr] = l;
			push(l);
		}

		if (cod == 2)
		{
			A[x] = -1;
			assert(Pos[x] != 0);
			assert(1<=x && x<=nr);
			push(Pos[x]);
			Pos[Heap[1]] = 0;
			Heap[1] = Heap[l--];
			Pos[Heap[1]] = 1;
			if (l>1) pop(1);
		}

		if (cod == 3) printf("%d\n", A[Heap[1]]);
	}
	return 0;
}