#include "towers.h"
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> pii;
int n;
vector<int> h;
set<int> st;
priority_queue<pair<int, pii> > pq;
vector<pii> q;
void init(int N, std::vector<int> H) {
n = N; h = H;
st.insert(0); st.insert(N - 1);
for(int i = 1; i < N - 1; ++i) {
if(h[i - 1] < h[i] && h[i] > h[i + 1]) st.insert(i);
if(h[i - 1] > h[i] && h[i] < h[i + 1]) st.insert(i);
}
for(auto it = st.begin(); it != st.end(); ++it) {
if(it == st.end()) break;
pq.push({-abs(H[*it] - H[*next(it)]), {*it, *next(it)}});
}
if(st.size() >= 2 && H[*st.begin()] > H[*next(st.begin())]) st.erase(st.begin());
if(st.size() >= 2 && H[*st.rbegin()] > H[*next(st.rbegin())]) st.erase(prev(st.end()));
q.push_back({0, (st.size() - 1) / 2 + 1});
while(!pq.empty()) {
int D = -pq.top().first;
while(!pq.empty() && -pq.top().first == D) {
int i = pq.top().second.first, j = pq.top().second.second;
pq.pop();
if(st.find(i) == st.end() || st.find(j) == st.end()) continue;
st.erase(st.find(i));
st.erase(st.find(j));
if(!st.empty() && *st.begin() < i && *st.rbegin() > j) {
int prv = *prev(st.lower_bound(i));
int nxt = *st.lower_bound(j);
pq.push({-abs(H[prv] - H[nxt]), {prv, nxt}});
}
}
if(st.size() >= 2 && H[*st.begin()] > H[*next(st.begin())]) st.erase(st.begin());
if(st.size() >= 2 && H[*st.rbegin()] > H[*next(st.rbegin())]) st.erase(prev(st.end()));
q.push_back({D + 1, (st.size() - 1) / 2 + 1});
}
}
int max_towers(int l, int r, int d) {
pii p = {d, -1};
int i = prev(upper_bound(q.begin(), q.end(), p)) - q.begin();
return q[i].second;
}
/*
7 3
10 20 60 40 50 30 70
1 5 10
2 2 100
0 6 17
7 3
10 20 60 40 50 30 70
0 6 10
0 6 100
0 6 17
*/
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
311 ms |
924 KB |
1st lines differ - on the 1st token, expected: '1', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
208 KB |
1st lines differ - on the 1st token, expected: '13', found: '131' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
208 KB |
1st lines differ - on the 1st token, expected: '13', found: '131' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
579 ms |
7028 KB |
1st lines differ - on the 1st token, expected: '11903', found: '33010' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
206 ms |
1888 KB |
Output is correct |
2 |
Incorrect |
789 ms |
7184 KB |
18774th lines differ - on the 1st token, expected: '16198', found: '16199' |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
208 KB |
1st lines differ - on the 1st token, expected: '13', found: '131' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
311 ms |
924 KB |
1st lines differ - on the 1st token, expected: '1', found: '2' |
2 |
Halted |
0 ms |
0 KB |
- |