Header Ad

HackerEarth Little Deepu and Array problem solution

In this HackerEarth Little Deepu and Array problem solution, Little Deepu loves positive things in life, positive girlfriends, positive marks. He's, in general a lover of positive things, so for obvious reasons he loves an array which contains positive elements.

Anyway, Little Deepu also thinks that not every value in his so called positive array deserves to be valued so high, so he makes an operation called HIT which takes exactly one argument, i.e., HIT (X). When HIT (X) is called, it decreases the value of all the elements of the array by 1 which are greater than X.

Now, to test how positive can you stay in life, he performs M HIT operations, and after they have been done, Deepu wants you to print the final array.


HackerEarth Little Deepu and Array problem solution


HackerEarth Little Deepu and Array problem solution.

#include<iostream>
#include<bits/stdc++.h>
using namespace std;
//
#include<string.h>
#include<math.h>


#define MAX 100005
#define inf 0x7fffffff

int N;
int arr[MAX];
int tree[4*MAX];
int lazy[4*MAX];

/**
* Build and init tree
*/
void build_tree(int node, int a, int b) {
if(a > b) return;

if(a == b) {
tree[node] = arr[a];
return;
}

build_tree(node*2, a, (a+b)/2);
build_tree(node*2+1, 1+(a+b)/2, b);

tree[node] = min(tree[node*2], tree[node*2+1]);
}

/**
* Increment elements within range [i, j] with value value
*/
void update_tree(int node, int a, int b, int i, int j, int x, int value) {



if(a > b || a > j || b < i) // Current segment is not within range [i, j]
return;

if(lazy[node] != 0) { // This node needs to be updated
tree[node] += lazy[node]; // Update it

if(a != b) {
lazy[node*2] += lazy[node]; // Mark child as lazy
lazy[node*2+1] += lazy[node]; // Mark child as lazy
}

lazy[node] = 0; // Reset it
}



if(tree[node] > x) { // Segment is fully within range
tree[node] += value;
if(a != b) { // Not leaf node
lazy[node*2] += value;
lazy[node*2+1] += value;
}
return;
}

if(a == b) return;

update_tree(node*2, a, (a+b)/2, i, j,x, value); // Updating left child
update_tree(1+node*2, 1+(a+b)/2, b, i, j,x, value); // Updating right child

tree[node] = min(tree[node*2], tree[node*2+1]); // Updating root with max value
}


/**
* Query tree to get max element value within range [i, j]
*/
int query_tree(int node, int a, int b, int i, int j) {

if(a > b || a > j || b < i) return inf;

if(lazy[node] != 0) {
tree[node] += lazy[node];

if(a != b) {
lazy[node*2] += lazy[node];
lazy[node*2+1] += lazy[node];
}

lazy[node] = 0;
}

if(a >= i && b <= j)
return tree[node];

int q1 = query_tree(node*2, a, (a+b)/2, i, j);
int q2 = query_tree(1+node*2, 1+(a+b)/2, b, i, j);
int res = min(q1, q2);

return res;
}

int main() {
cin >> N;
assert(N < 1000001);
vector<pair<int,int> > ds;
for(int i=0;i<N;i++){
int a;
cin >> a;
assert(a < 1000000000);
ds.push_back(make_pair(a,i+1));
}
sort(ds.begin(),ds.end());
for(int i=0;i<N;i++){
arr[i] = ds[i].first;
}
build_tree(1, 0, N-1);
memset(lazy, 0, sizeof lazy);
int Q;
cin >> Q;
assert(Q <= 20000);
while(Q--){
int x;
cin >> x;
assert(x <= 1000000000);
update_tree(1, 0, N-1 , 0 , N-1, x, -1);
}
for(int i=0;i<N;i++){
ds[i].first = query_tree(1, 0, N-1, i, i);
swap(ds[i].first, ds[i].second);
}
sort(ds.begin(),ds.end());
for(int i=0;i<N;i++){
cout << ds[i].second << " ";
}
cout << endl;
}


Post a Comment

0 Comments