Massive Matrices

#include <iostream>

using namespace std;
typedef long long ll;

const int N = (1<<18);

int n;
ll at[200005], bt[200005];

ll st[2*N];

ll res = 0;

void muuta(int k, ll x) {
	k += N;
	st[k] = x;
	for (k /= 2; k >= 1; k /= 2) {
		st[k] = max(st[2*k], st[2*k+1]);
	}
}

ll rmq(int a, int b) {
	a += N;
	b += N;
	ll cres = -1;
	while (a <= b) {
		if (a%2 == 1) {
			cres = max(cres, st[a++]);
		}
		if (b%2 == 0) {
			cres = max(cres, st[b--]);
		}
		a /= 2;
		b /= 2;
	}
	return cres;
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cin >> n;
	for (int i = 1; i <= n; ++i) {
		cin >> at[i];
	}
	for (int i = 1; i <= n; ++i) {
		cin >> bt[i];
		muuta(i, bt[i]);
	}
	for (int i = 1; i <= n; ++i) {
		if (at[i] == 0) {
			continue;
		}
		ll cmx = rmq(1, at[i]);
		res += cmx;
	}
	cout << res << "\n";
	return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: ACCEPTED

Test 24

Verdict: ACCEPTED

Test 25

Verdict: ACCEPTED

Test 26

Verdict: ACCEPTED

Test 27

Verdict: ACCEPTED

Test 28

Verdict: ACCEPTED

Test 29

Verdict: ACCEPTED

Test 30

Verdict: ACCEPTED

Test 31

Verdict: ACCEPTED

Test 32

Verdict: ACCEPTED

Test 33

Verdict: ACCEPTED