Pagini recente » Cod sursa (job #657926) | Cod sursa (job #794856) | Cod sursa (job #1530049) | Cod sursa (job #2009083) | Cod sursa (job #229890)
Cod sursa(job #229890)
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
const char iname[] = "schi.in";
const char oname[] = "schi.out";
#define getc(n) (n->cnt = (!n->el) + (n->l != NULL ? n->l->cnt : 0) + (n->r != NULL ? n->r->cnt : 0))
struct T {
int key, cnt, el;
T *l, *r;
T(int key) {
this->key = key, l = r = NULL;
cnt = el = 0;
}
} *R;
void insert(T* &n, int lo, int hi)
{
if (hi < lo) return ;
int mid = (hi + lo) >> 1;
n = new T(mid);
insert(n->l, lo, mid - 1);
insert(n->r, mid + 1, hi);
getc(n);
}
void lookup(T* &n, int cnt, int el)
{
if (cnt <= (n->l != NULL ? n->l->cnt : 0))
lookup(n->l, cnt, el);
else {
if ((n->l != NULL ? n->l->cnt : 0) + (!n->el) == cnt)
n->el = el;
else
lookup(n->r, cnt - (n->l != NULL ? n->l->cnt : 0) - (!n->el), el);
}
getc(n);
}
void print(T* &n, ofstream& out) {
if (n == NULL) return ;
print(n->l, out), out << n->el <<'\n', print(n->r, out);
}
int main(void)
{
srand(unsigned(time(NULL)));
R = NULL;
ifstream in(iname); ofstream out(oname);
vector <int> V;
int n;
in >> n;
V.resize(n);
for (int i = 0; i < n; ++ i) in >> V[i];
insert(R, 0, n - 1);
for (int i = n - 1; i >= 0; -- i) lookup(R, V[i], i + 1);
print(R, out);
return 0;
}