#include "jumps.h"
#include<bits/stdc++.h>
using namespace std;
#define INF 1000000000
#define MAXN 200000
#define MAXH 200000
#define MAXLG 17
int n, prevHigher[MAXN + 3], nextHigher[MAXN + 3], highest[MAXH + 3][MAXLG + 3], secondHighest[MAXH + 3][MAXLG + 3];
int spTableMax[MAXN + 3][MAXLG + 3];
vector<int> h;
void initPrevHigher() {
stack<int> stHeight;
for (int i = 0; i < n; i++) {
while (!stHeight.empty() && h[i] > stHeight.top()) {
stHeight.pop();
}
prevHigher[i] = (stHeight.empty() ? INF : stHeight.top());
stHeight.push(h[i]);
}
}
void initNextHigher() {
stack<int> stHeight;
for (int i = n - 1; i >= 0; i--) {
while (!stHeight.empty() && h[i] > stHeight.top()) {
stHeight.pop();
}
nextHigher[i] = (stHeight.empty() ? INF : stHeight.top());
stHeight.push(h[i]);
}
}
void initHighestAndSecondHighest() {
for (int h = 1; h <= n; h++) {
highest[h][0] = secondHighest[h][0] = INF;
}
for (int i = 0; i < n; i++) {
highest[h[i]][0] = max(prevHigher[i], nextHigher[i]);
secondHighest[h[i]][0] = min(prevHigher[i], nextHigher[i]);
}
for (int lg = 1; lg <= MAXLG; lg++) {
for (int h = 1; h <= n; h++) {
if (highest[h][lg - 1] < INF) {
highest[h][lg] = highest[highest[h][lg - 1]][lg - 1];
} else {
highest[h][lg] = INF;
}
if (secondHighest[h][lg - 1] < INF) {
secondHighest[h][lg] = secondHighest[secondHighest[h][lg - 1]][lg - 1];
} else {
secondHighest[h][lg] = INF;
}
}
}
}
int getLastHeightFollowingHighest(int startingHeight, int numJumps) {
int currentHeight = startingHeight;
for (int lg = 0; lg <= MAXLG && currentHeight < INF; lg++) {
if (numJumps & (1 << lg)) {
currentHeight = highest[currentHeight][lg];
}
}
return currentHeight;
}
int getLastHeightFollowingSecondHighest(int startingHeight, int numJumps) {
int currentHeight = startingHeight;
for (int lg = 0; lg <= MAXLG && currentHeight < INF; lg++) {
if (numJumps & (1 << lg)) {
currentHeight = secondHighest[currentHeight][lg];
}
}
return currentHeight;
}
void initSpTableMax() {
for (int i = 0; i < n; i++) {
spTableMax[i][0] = h[i];
}
for (int lg = 1; lg <= MAXLG; lg++) {
for (int i = 0; i < n; i++) {
if (i + (1 << (lg - 1)) < n) {
spTableMax[i][lg] = max(spTableMax[i][lg - 1], spTableMax[i + (1 << (lg - 1))][lg - 1]);
} else {
spTableMax[i][lg] = spTableMax[i][lg - 1];
}
}
}
}
int rangeMaxQuery(int a, int b) {
int len = b - a + 1;
int msb;
for (int lg = MAXLG; lg >= 0; lg--) {
if (len & (1 << lg)) {
msb = lg;
break;
}
}
// cout << "hihi " << msb << " " << a << " " << b + 1 - (1 << msb) << endl;
return max(spTableMax[a][msb], spTableMax[b + 1 - (1 << msb)][msb]);
}
void init(int N, std::vector<int> H) {
n = N;
h = H;
initPrevHigher();
initNextHigher();
initHighestAndSecondHighest();
initSpTableMax();
// cout << "\nprevHigher\n";
// for (int i = 0; i < n; i++) {
// cout << prevHigher[i] << endl;
// }
// cout << "\nnextHigher\n";
// for (int i = 0; i < n; i++) {
// cout << nextHigher[i] << endl;
// }
// cout << endl;
// for (int i = 1; i <= 3; i++) {
// cout << "highest[" << i << "]: " << highest[i][0] << endl;
// cout << "secondHighest[" << i << "]: " << secondHighest[i][0] << endl;
// }
}
int minimum_jumps(int A, int B, int C, int D) {
int startingHeight = rangeMaxQuery(A, B);
// cout << "\nstartingHeight: " << startingHeight << endl;
// Calculate max number of jumps if we follow the highest pointer
int numHighestJumps;
if (true) {
int lo = 0, hi = MAXN, mid;
while (hi >= lo) {
mid = (lo + hi)/2;
if (getLastHeightFollowingHighest(startingHeight, mid) <= h[C]) {
lo = mid + 1;
} else {
hi = mid - 1;
}
}
numHighestJumps = hi;
}
// cout << "\nnumHighestJumps: " << numHighestJumps << endl;
int heightAfterFollowingHighest = getLastHeightFollowingHighest(startingHeight, numHighestJumps);
// cout << "\nheight after following highest:" << heightAfterFollowingHighest << endl;
// Calculate max number of jumps if we follow the second highest pointer
int numSecondHighestJumps;
if (true) {
int lo = 0, hi = MAXN, mid;
while (hi >= lo) {
mid = (lo + hi)/2;
if (getLastHeightFollowingSecondHighest(heightAfterFollowingHighest, mid) <= h[C]) {
lo = mid + 1;
} else {
hi = mid - 1;
}
}
numSecondHighestJumps = hi;
}
int finalHeight = getLastHeightFollowingSecondHighest(heightAfterFollowingHighest, numSecondHighestJumps);
// cout << "numSecondHighestJumps: " << numSecondHighestJumps << endl;
// cout << "finalHeight: " << finalHeight << endl;
if (finalHeight == h[C]) {
return numHighestJumps + numSecondHighestJumps;
} else {
return -1;
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
336 KB |
Output is correct |
3 |
Correct |
225 ms |
40916 KB |
Output is correct |
4 |
Correct |
1443 ms |
51232 KB |
Output is correct |
5 |
Correct |
1375 ms |
25996 KB |
Output is correct |
6 |
Correct |
1620 ms |
51332 KB |
Output is correct |
7 |
Correct |
1187 ms |
35092 KB |
Output is correct |
8 |
Correct |
1618 ms |
51268 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
208 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
0 ms |
208 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
208 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
336 KB |
Output is correct |
2 |
Correct |
1 ms |
208 KB |
Output is correct |
3 |
Correct |
1 ms |
336 KB |
Output is correct |
4 |
Correct |
263 ms |
23588 KB |
Output is correct |
5 |
Correct |
1129 ms |
51112 KB |
Output is correct |
6 |
Correct |
640 ms |
8784 KB |
Output is correct |
7 |
Correct |
1361 ms |
51128 KB |
Output is correct |
8 |
Correct |
753 ms |
17736 KB |
Output is correct |
9 |
Correct |
1162 ms |
51144 KB |
Output is correct |
10 |
Correct |
1499 ms |
51296 KB |
Output is correct |
11 |
Correct |
1415 ms |
51256 KB |
Output is correct |
12 |
Correct |
1435 ms |
51216 KB |
Output is correct |
13 |
Correct |
1263 ms |
51148 KB |
Output is correct |
14 |
Correct |
1679 ms |
51556 KB |
Output is correct |
15 |
Correct |
1065 ms |
51880 KB |
Output is correct |
16 |
Correct |
1053 ms |
51896 KB |
Output is correct |
17 |
Correct |
0 ms |
208 KB |
Output is correct |
18 |
Correct |
0 ms |
336 KB |
Output is correct |
19 |
Correct |
2 ms |
336 KB |
Output is correct |
20 |
Correct |
4 ms |
336 KB |
Output is correct |
21 |
Correct |
2 ms |
336 KB |
Output is correct |
22 |
Correct |
1 ms |
336 KB |
Output is correct |
23 |
Correct |
2 ms |
336 KB |
Output is correct |
24 |
Correct |
3 ms |
336 KB |
Output is correct |
25 |
Correct |
0 ms |
336 KB |
Output is correct |
26 |
Correct |
1 ms |
464 KB |
Output is correct |
27 |
Correct |
21 ms |
720 KB |
Output is correct |
28 |
Correct |
23 ms |
848 KB |
Output is correct |
29 |
Correct |
21 ms |
720 KB |
Output is correct |
30 |
Correct |
20 ms |
720 KB |
Output is correct |
31 |
Correct |
18 ms |
720 KB |
Output is correct |
32 |
Correct |
1 ms |
208 KB |
Output is correct |
33 |
Correct |
65 ms |
29684 KB |
Output is correct |
34 |
Correct |
115 ms |
51176 KB |
Output is correct |
35 |
Correct |
107 ms |
51216 KB |
Output is correct |
36 |
Correct |
118 ms |
51148 KB |
Output is correct |
37 |
Correct |
117 ms |
51484 KB |
Output is correct |
38 |
Correct |
106 ms |
52000 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
336 KB |
Output is correct |
2 |
Correct |
1 ms |
208 KB |
Output is correct |
3 |
Correct |
1 ms |
336 KB |
Output is correct |
4 |
Correct |
263 ms |
23588 KB |
Output is correct |
5 |
Correct |
1129 ms |
51112 KB |
Output is correct |
6 |
Correct |
640 ms |
8784 KB |
Output is correct |
7 |
Correct |
1361 ms |
51128 KB |
Output is correct |
8 |
Correct |
753 ms |
17736 KB |
Output is correct |
9 |
Correct |
1162 ms |
51144 KB |
Output is correct |
10 |
Correct |
1499 ms |
51296 KB |
Output is correct |
11 |
Correct |
1415 ms |
51256 KB |
Output is correct |
12 |
Correct |
1435 ms |
51216 KB |
Output is correct |
13 |
Correct |
1263 ms |
51148 KB |
Output is correct |
14 |
Correct |
1679 ms |
51556 KB |
Output is correct |
15 |
Correct |
1065 ms |
51880 KB |
Output is correct |
16 |
Correct |
1053 ms |
51896 KB |
Output is correct |
17 |
Correct |
0 ms |
208 KB |
Output is correct |
18 |
Correct |
0 ms |
336 KB |
Output is correct |
19 |
Correct |
2 ms |
336 KB |
Output is correct |
20 |
Correct |
4 ms |
336 KB |
Output is correct |
21 |
Correct |
2 ms |
336 KB |
Output is correct |
22 |
Correct |
1 ms |
336 KB |
Output is correct |
23 |
Correct |
2 ms |
336 KB |
Output is correct |
24 |
Correct |
3 ms |
336 KB |
Output is correct |
25 |
Correct |
0 ms |
336 KB |
Output is correct |
26 |
Correct |
1 ms |
464 KB |
Output is correct |
27 |
Correct |
21 ms |
720 KB |
Output is correct |
28 |
Correct |
23 ms |
848 KB |
Output is correct |
29 |
Correct |
21 ms |
720 KB |
Output is correct |
30 |
Correct |
20 ms |
720 KB |
Output is correct |
31 |
Correct |
18 ms |
720 KB |
Output is correct |
32 |
Correct |
1 ms |
208 KB |
Output is correct |
33 |
Correct |
65 ms |
29684 KB |
Output is correct |
34 |
Correct |
115 ms |
51176 KB |
Output is correct |
35 |
Correct |
107 ms |
51216 KB |
Output is correct |
36 |
Correct |
118 ms |
51148 KB |
Output is correct |
37 |
Correct |
117 ms |
51484 KB |
Output is correct |
38 |
Correct |
106 ms |
52000 KB |
Output is correct |
39 |
Correct |
0 ms |
336 KB |
Output is correct |
40 |
Correct |
0 ms |
208 KB |
Output is correct |
41 |
Correct |
0 ms |
336 KB |
Output is correct |
42 |
Correct |
306 ms |
23516 KB |
Output is correct |
43 |
Correct |
1203 ms |
51112 KB |
Output is correct |
44 |
Correct |
713 ms |
8656 KB |
Output is correct |
45 |
Correct |
1050 ms |
51148 KB |
Output is correct |
46 |
Correct |
380 ms |
17844 KB |
Output is correct |
47 |
Correct |
1199 ms |
51144 KB |
Output is correct |
48 |
Correct |
1527 ms |
51200 KB |
Output is correct |
49 |
Correct |
1108 ms |
51348 KB |
Output is correct |
50 |
Correct |
1658 ms |
51308 KB |
Output is correct |
51 |
Correct |
1131 ms |
51272 KB |
Output is correct |
52 |
Correct |
1445 ms |
51480 KB |
Output is correct |
53 |
Correct |
1061 ms |
51908 KB |
Output is correct |
54 |
Correct |
774 ms |
51964 KB |
Output is correct |
55 |
Correct |
0 ms |
336 KB |
Output is correct |
56 |
Incorrect |
172 ms |
50884 KB |
Output isn't correct |
57 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
336 KB |
Output is correct |
3 |
Correct |
225 ms |
40916 KB |
Output is correct |
4 |
Correct |
1443 ms |
51232 KB |
Output is correct |
5 |
Correct |
1375 ms |
25996 KB |
Output is correct |
6 |
Correct |
1620 ms |
51332 KB |
Output is correct |
7 |
Correct |
1187 ms |
35092 KB |
Output is correct |
8 |
Correct |
1618 ms |
51268 KB |
Output is correct |
9 |
Incorrect |
0 ms |
208 KB |
Output isn't correct |
10 |
Halted |
0 ms |
0 KB |
- |