Cod sursa(job #716182)

Utilizator gener.omerGener Omer gener.omer Data 18 martie 2012 13:48:32
Problema Heapuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.66 kb
#include <stdio.h>
#include <assert.h>

#define maxn 200010

int N, L, NR;
int A[maxn], Heap[maxn], Pos[maxn];

#define SWAP(a, b){\
    a = a + b;\
    b = a - b;\
    a = a - b;\
}

inline int getLeft(int p){ return 2 * p;}
inline int getRight(int p){ return 2 * p + 1;}
inline int getParent(int p){ return p / 2;}
inline int getMin(){ return A[Heap[1]];}

void push(int p){
	while(p > 1 && (A[Heap[p]] < A[Heap[getParent(p)]])){
		SWAP(Heap[p], Heap[getParent(p)]);
		Pos[Heap[p]] = p;
		Pos[Heap[getParent(p)]] = getParent(p);
		p = getParent(p);
	}
}

void pop(int p){
	int son;
	do{
		son = 0;
		if(getLeft(p) <= L){
			son = getLeft(p);
			if(getRight(p) <= L && A[Heap[son]] > A[Heap[getRight(p)]])
				son = getRight(p);
		}
		if(son){
			if(A[Heap[son]] < A[Heap[p]]){
				SWAP(Heap[son], Heap[p])
				Pos[Heap[son]] = son;
				Pos[Heap[p]] = p;
				p = son;
			}
			else
				son = 0;
		}
	}while(son);
}

void insert(int key){
	/*++NR, ++L;
	A[NR] = key;
	Heap[L] = N;
	Pos[N] = N;

	up(N);*/
}

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

	int i, x, cod;

	scanf("%d ", &N);

	for (i=1; i<=N; i++)
	{
		scanf("%d ", &cod);

		if (cod == 1)
		{
			scanf("%d", &x);
			insert(x);

			L++, NR++;
			A[NR] = x;
			Heap[L] = NR;
			Pos[NR] = L;

			push(L);
		}

		if (cod == 2)
		{
			scanf("%d", &x);
			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", getMin());
		}
	}

	return 0;
}