Cod sursa(job #826786)

Utilizator UnforgivenMihai Catalin Botezatu Unforgiven Data 1 decembrie 2012 11:20:06
Problema Schi Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.96 kb
#include <fstream>

#define MAX_SIZE 1 << 16
#define left_son ((nod) << 1)
#define right_son ((nod) << 1) +1
using namespace std;


typedef struct
{
    int value;
    int pos;
}st;

st tree[MAX_SIZE];
int vect[MAX_SIZE];


int A;
int B;
int val;

void build_tree(int nod , int left , int right)
{
    if (left == right)
    {
        tree[nod].value = vect[left];
        tree[nod].pos = left;
        vect[left] = 0;
        return;
    }
    int middle = (left + right) >> 1;
    build_tree(left_son , left , middle);
    build_tree(right_son , middle+1 , right);
    if (tree[right_son].value <= tree[left_son].value)
    {
        tree[nod] = tree[right_son];
    }
    else
    {
        tree[nod] = tree[left_son];
    }
}

void update_tree(int nod , int left , int right)
{
    if (A > B) return;
    if (A <=left &&  right <= B)
    {
        tree[nod].value += val;
        vect[nod] += val;
        return;
    }
    int middle = (left + right) >> 1;
    if (A <= middle) update_tree(left_son , left , middle);
    if (middle < B) update_tree(right_son ,middle+1,right);

    if (tree[right_son].value <= tree[left_son].value)
    {
        tree[nod].value  = tree[right_son].value + vect[nod];
        tree[nod].pos = tree[right_son].pos;
    }
    else
    {
        tree[nod].value  = tree[left_son].value + vect[nod];
        tree[nod].pos = tree[left_son].pos;
    }
}

int main()
{
    ifstream input("schi.in");
    ofstream output("schi.out");
    int N;
    input >> N;
    for (int i = 1;i<=N;i++)
    {
        input >> vect[i];
    }
    build_tree(1,1,N);
    for (int i = 1 ; i<=N;i++)
    {
        output << tree[1].pos << "\n";
        A = 1;
        B = tree[1].pos-1;
        val = 1;
        update_tree(1,1,N);

        A = tree[1].pos;
        B = tree[1].pos;
        val = MAX_SIZE;

        update_tree(1,1,N);


    }
    input.close();
    output.close();

    return 0;
}