#include <bits/stdc++.h>
#pragma GCC Optimize("O3")
#define FOR(i, x, y) for (ll i = x; i < y; i++)
#define MOD 1000000007
typedef long long ll;
using namespace std;
ll h[500000], dp[500000];
double calc(ll k, ll i) { return h[k] + sqrt(abs(i - k)); }
int main() {
iostream::sync_with_stdio(false);
cin.tie(0);
ll n;
cin >> n;
FOR(i, 0, n) {
cin >> h[i];
dp[i] = h[i];
}
deque<ll> q;
q.push_back(0);
FOR(i, 1, n) {
while (q.size() > 1 && calc(q[0], i) <= calc(q[1], i)) q.pop_front();
dp[i] = max(dp[i], (ll)ceil(calc(q[0], i)));
while (q.size() > 1 &&
calc(q.back(), i + 1) - calc(q[q.size() - 2], i + 1) <=
calc(i, i + 1) - calc(q.back(), i + 1))
q.pop_back();
q.push_back(i);
}
q.clear();
q.push_back(n - 1);
for (ll i = n - 2; i >= 0; i--) {
while (q.size() > 1 && calc(q[0], i) <= calc(q[1], i)) q.pop_front();
dp[i] = max(dp[i], (ll)ceil(calc(q[0], i)));
while (q.size() > 1 &&
calc(q.back(), i + 1) - calc(q[q.size() - 2], i + 1) <=
calc(i, i + 1) - calc(q.back(), i + 1))
q.pop_back();
q.push_back(i);
}
FOR(i, 0, n) cout << dp[i] - h[i] << '\n';
return 0;
}
Compilation message
pio.cpp:2:0: warning: ignoring #pragma GCC Optimize [-Wunknown-pragmas]
#pragma GCC Optimize("O3")
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
376 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
13 ms |
1144 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
20 ms |
1656 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
23 ms |
1724 KB |
Output is correct |
2 |
Correct |
16 ms |
2168 KB |
Output is correct |
3 |
Incorrect |
21 ms |
1868 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
35 ms |
2296 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
80 ms |
4952 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
129 ms |
9336 KB |
Output is correct |
2 |
Correct |
108 ms |
7288 KB |
Output is correct |
3 |
Incorrect |
107 ms |
9340 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
182 ms |
13176 KB |
Output is correct |
2 |
Correct |
153 ms |
10104 KB |
Output is correct |
3 |
Incorrect |
153 ms |
13176 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
164 ms |
10744 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |