#include <bits/stdc++.h>
using namespace std;
#define f first
#define s second
#define pb push_back
#define mp make_pair
#define all(v) v.begin(), v.end()
#define sz(v) (int)v.size()
#define MOO(i, a, b) for(int i=a; i<b; i++)
#define M00(i, a) for(int i=0; i<a; i++)
#define MOOd(i,a,b) for(int i = (b)-1; i >= a; i--)
#define M00d(i,a) for(int i = (a)-1; i>=0; i--)
#define FAST ios::sync_with_stdio(0); cin.tie(0);
#define finish(x) return cout << x << '\n', 0;
#define dbg(x) cerr << ">>> " << #x << " = " << x << "\n";
#define _ << " _ " <<
typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef pair<int,int> pi;
typedef pair<ld,ld> pd;
typedef complex<ld> cd;
const int MAX_N = 500500;
int n;
ld eval(ld x, pair<ll,ll> curve) {
return curve.s + sqrt(curve.f - x);
}
ll evalLong(ll x, pair<ll,ll> curve) { // ceil(eval(x, curve))
ld y = eval(x,curve);
ll y1 = y;
if((y1-curve.s)*(y1-curve.s) >= curve.f-x) return y1;
return y1+1;
}
ld intersect(pair<ll,ll> c1, pair<ll,ll> c2) { // c2 comes after c1
if(eval(c1.f, c2) <= c1.s) return c1.f;
if(eval(0, c2) >= eval(0, c1)) return -1;
ld a = c1.s; ld b = c1.f; ld c = c2.s; ld d = c2.f;
ld res = -a*a*a*a+4*a*a*a*c+2*a*a*b-6*a*a*c*c+2*a*a*d-4*a*b*c+4*a*c*c*c-4*a*c*d-b*b+2*b*c*c+2*b*d-c*c*c*c+2*c*c*d-d*d;
res /= 4*(a-c)*(a-c);
return res;
}
vector<ll> solve(vector<ll> v) {
deque<pair<ll,ll>> dq;
MOO(i, 1, n) {
pair<ll,ll> curve = mp(i, v[i]);
while(1) {
if(dq.empty()) break;
else if(intersect(dq[sz(dq)-1], curve) < 0) dq.pop_back();
else {
if(sz(dq) >= 2) {
if(intersect(dq[sz(dq)-2], curve) < intersect(dq[sz(dq)-2], dq[sz(dq)-1])) {
dq.pop_back();
} else {
break;
}
} else {
break;
}
}
}
dq.pb(curve);
}
vector<ll> ans;
M00(i, n-1) {
while(dq[0].f <= i) dq.pop_front();
while(sz(dq) >= 2 && eval(i, dq[1]) > eval(i, dq[0])) dq.pop_front();
ans.pb(max(0LL,evalLong(i, dq[0])-v[i]));
}
ans.pb(-1);
return ans;
}
int main() { FAST
cin >> n;
vector<ll> arr(n);
M00(i, n) cin >> arr[i];
vector<ll> res1 = solve(arr);
reverse(all(arr));
vector<ll> res2 = solve(arr);
reverse(all(res2));
M00(i, n) {
cout << max(res1[i], res2[i]) << "\n";
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
4 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
384 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
18 ms |
1556 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
26 ms |
2420 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
37 ms |
2944 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
51 ms |
4464 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
115 ms |
9584 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
170 ms |
15520 KB |
Output is correct |
2 |
Correct |
143 ms |
18852 KB |
Output is correct |
3 |
Incorrect |
156 ms |
17572 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
230 ms |
20124 KB |
Output is correct |
2 |
Correct |
201 ms |
25116 KB |
Output is correct |
3 |
Incorrect |
207 ms |
22936 KB |
Output isn't correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
226 ms |
20200 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |