#include "secret.h"
#include <bits/stdc++.h>
using namespace std;
const int nmax = 1e3 + 5;
vector<int> v;
namespace RMQ {
int atr[nmax];
int rmq[12][nmax];
void color(int l, int r, int bit = 0) {
//cerr << l << ' ' << r << ' ' << bit << '\n';
if(l + 1 >= r) {
while(l <= r)
atr[l] = bit, l++;
return;
}
int mid = l + r >> 1;
atr[mid] = atr[mid + 1] = bit;
color(l, mid - 1, bit + 1);
color(mid + 2, r, bit + 1);
rmq[bit][mid] = v[mid];
rmq[bit][mid + 1] = v[mid + 1];
for(int i = mid - 1; i >= l; i--)
rmq[bit][i] = Secret(v[i], rmq[bit][i + 1]);
for(int i = mid + 2; i <= r; i++)
rmq[bit][i] = Secret(rmq[bit][i - 1], v[i]);
return;
}
namespace IRMQ {
pair<int,int> irmq[12][nmax];
int lg[nmax];
void init(int n, int *vec) {
lg[1] = 0;
for(int i = 2; i <= n; i++)
lg[i] = lg[i >> 1] + 1;
for(int i = 0; i < n; i++)
irmq[0][i] = {vec[i], i};
for(int bit = 1; bit < 12; bit++) {
for(int i = 0; i + (1 << bit) <= n; i++)
irmq[bit][i] = min(irmq[bit - 1][i], irmq[bit - 1][i + (1 << bit - 1)]);
}
}
pair<int,int> query(int l, int r) {
int st = lg[r - l + 1];
return min(irmq[st][l], irmq[st][r - (1 << st) + 1]);
}
}
int query(int l, int r) {
if(l == r)
return v[l];
else if(l == r - 1)
return Secret(v[l], v[r]);
auto level = IRMQ::query(l, r);
if(r == level.second)
return rmq[level.first][l];
if(l == level.second + 1)
return rmq[level.first][r];
return Secret(rmq[level.first][l], rmq[level.first][r]);
}
}
void Init(int N, int A[]) {
v.resize(N);
for(int i = 0; i < N; i++)
v[i] = A[i];
RMQ::color(0, N - 1);
RMQ::IRMQ::init(N, RMQ::atr);
}
int Query(int L, int R) {
return RMQ::query(L, R);
}
Compilation message
secret.cpp: In function 'void RMQ::color(int, int, int)':
secret.cpp:18:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
18 | int mid = l + r >> 1;
| ~~^~~
secret.cpp: In function 'void RMQ::IRMQ::init(int, int*)':
secret.cpp:41:76: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
41 | irmq[bit][i] = min(irmq[bit - 1][i], irmq[bit - 1][i + (1 << bit - 1)]);
| ~~~~^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
113 ms |
2528 KB |
Wrong Answer: Query(128, 254) - expected : 576149945, actual : 826360691. |
2 |
Incorrect |
120 ms |
2556 KB |
Wrong Answer: Query(236, 238) - expected : 173116186, actual : 416517106. |
3 |
Incorrect |
112 ms |
2464 KB |
Wrong Answer: Query(128, 153) - expected : 959658850, actual : 219466354. |
4 |
Incorrect |
457 ms |
4468 KB |
Wrong Answer: Query(750, 990) - expected : 629178548, actual : 554844227. |
5 |
Incorrect |
438 ms |
4372 KB |
Wrong Answer: Query(626, 678) - expected : 512755800, actual : 356156688. |
6 |
Incorrect |
445 ms |
4464 KB |
Wrong Answer: Query(250, 258) - expected : 907404, actual : 12475405. |
7 |
Correct |
469 ms |
4380 KB |
Output is correct - number of calls to Secret by Init = 6996, maximum number of calls to Secret by Query = 1 |
8 |
Correct |
478 ms |
4564 KB |
Output is correct - number of calls to Secret by Init = 6996, maximum number of calls to Secret by Query = 1 |
9 |
Correct |
470 ms |
4416 KB |
Output is correct - number of calls to Secret by Init = 6996, maximum number of calls to Secret by Query = 1 |
10 |
Correct |
481 ms |
4460 KB |
Output is correct - number of calls to Secret by Init = 6996, maximum number of calls to Secret by Query = 1 |