| # |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
| 646740 |
2022-09-30T13:56:53 Z |
inksamurai |
Krov (COCI17_krov) |
C++17 |
|
1500 ms |
22116 KB |
#include <bits/stdc++.h>
// cut here
#ifdef _MSC_VER
#include <intrin.h>
#endif
namespace atcoder {
namespace internal {
int ceil_pow2(int n) {
int x = 0;
while ((1U << x) < (unsigned int)(n)) x++;
return x;
}
int bsf(unsigned int n) {
#ifdef _MSC_VER
unsigned long index;
_BitScanForward(&index, n);
return index;
#else
return __builtin_ctz(n);
#endif
}
} // namespace internal
} // namespace atcoder
namespace atcoder {
template <class S, S (*op)(S, S), S (*e)()> struct segtree {
public:
segtree() : segtree(0) {}
segtree(int n) : segtree(std::vector<S>(n, e())) {}
segtree(const std::vector<S>& v) : _n(int(v.size())) {
log = internal::ceil_pow2(_n);
size = 1 << log;
d = std::vector<S>(2 * size, e());
for (int i = 0; i < _n; i++) d[size + i] = v[i];
for (int i = size - 1; i >= 1; i--) {
update(i);
}
}
void set(int p, S x) {
assert(0 <= p && p < _n);
p += size;
d[p] = x;
for (int i = 1; i <= log; i++) update(p >> i);
}
S get(int p) {
assert(0 <= p && p < _n);
return d[p + size];
}
S prod(int l, int r) {
assert(0 <= l && l <= r && r <= _n);
S sml = e(), smr = e();
l += size;
r += size;
while (l < r) {
if (l & 1) sml = op(sml, d[l++]);
if (r & 1) smr = op(d[--r], smr);
l >>= 1;
r >>= 1;
}
return op(sml, smr);
}
S all_prod() { return d[1]; }
template <bool (*f)(S)> int max_right(int l) {
return max_right(l, [](S x) { return f(x); });
}
template <class F> int max_right(int l, F f) {
assert(0 <= l && l <= _n);
assert(f(e()));
if (l == _n) return _n;
l += size;
S sm = e();
do {
while (l % 2 == 0) l >>= 1;
if (!f(op(sm, d[l]))) {
while (l < size) {
l = (2 * l);
if (f(op(sm, d[l]))) {
sm = op(sm, d[l]);
l++;
}
}
return l - size;
}
sm = op(sm, d[l]);
l++;
} while ((l & -l) != l);
return _n;
}
template <bool (*f)(S)> int min_left(int r) {
return min_left(r, [](S x) { return f(x); });
}
template <class F> int min_left(int r, F f) {
assert(0 <= r && r <= _n);
assert(f(e()));
if (r == 0) return 0;
r += size;
S sm = e();
do {
r--;
while (r > 1 && (r % 2)) r >>= 1;
if (!f(op(d[r], sm))) {
while (r < size) {
r = (2 * r + 1);
if (f(op(d[r], sm))) {
sm = op(d[r], sm);
r--;
}
}
return r + 1 - size;
}
sm = op(d[r], sm);
} while ((r & -r) != r);
return 0;
}
private:
int _n, size, log;
std::vector<S> d;
void update(int k) { d[k] = op(d[2 * k], d[2 * k + 1]); }
};
} // namespace atcoder
// cut here
using namespace std;
#define int ll
#define rep(i,n) for(int i=0;i<n;i++)
#define rng(i,c,n) for(int i=c;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define fi first
#define se second
#define pb push_back
#define sz(a) (int)a.size()
#define vec(...) vector<__VA_ARGS__>
#define _3SJjfN1 ios::sync_with_stdio(0),cin.tie(0)
typedef long long ll;
using pii=pair<int,int>;
using vi=vector<int>;
void print(){cout<<'\n';}
template<class h,class...t>
void print(const h&v,const t&...u){cout<<v<<' ',print(u...);}
// e
struct coordinates{
vi xs;
void clear(){
xs.clear();
}
void add(int x){
xs.pb(x);
}
void affine(){
sort(xs.begin(),xs.end());
xs.erase(unique(xs.begin(),xs.end()),xs.end());
}
};
int op(int l,int r){return l+r;}
int e(){return 0;}
signed main(){
_3SJjfN1;
int n;
cin>>n;
vi a(n);
rep(i,n){
cin>>a[i];
}
// to the left i need
// x - pvt = y - i
// to the right i need
// x - y = i - pvt
// x + pvt = i + y
coordinates tmp;
rep(i,n){
tmp.add(a[i]-i);
tmp.add(a[i]+i);
}
tmp.affine();
const int m=sz(tmp.xs);
vi idsl(n),idsr(n);
rep(i,n){
{
int v=a[i]-i;
int j=lower_bound(tmp.xs.begin(),tmp.xs.end(),v)-tmp.xs.begin();
idsl[i]=j;
}
{
int v=a[i]+i;
int j=lower_bound(tmp.xs.begin(),tmp.xs.end(),v)-tmp.xs.begin();
idsr[i]=j;
}
}
atcoder::segtree<int,op,e> segl(m),cntl(m),segr(m),cntr(m);
int psunl=0,pcntl=0,psunr=0,pcntr=0;
auto af=[&](const int&pvt,const int&x)->int{
int res=0;
// to the left
int v=x-pvt;
// >=
int j=lower_bound(tmp.xs.begin(),tmp.xs.end(),v)-tmp.xs.begin();
int sun=psunl,now=pcntl;
if(j>=0 and j<sz(tmp.xs)){
int vala=segl.prod(j,m);
int valb=cntl.prod(j,m);
res+=vala-valb*v;
sun-=vala;
now-=valb;
}
// <=
if(j>=0 and j<=sz(tmp.xs)){
res+=now*v-sun;
}
// to the right
v=x+pvt;
j=lower_bound(tmp.xs.begin(),tmp.xs.end(),v)-tmp.xs.begin();
sun=psunr,now=pcntr;
if(j>=0 and j<sz(tmp.xs)){
int vala=segr.prod(j,m);
int valb=cntr.prod(j,m);
res+=vala-valb*v;
sun-=vala;
now-=valb;
}
// <=
if(j>=0 and j<=sz(tmp.xs)){
res+=now*v-sun;
}
return res;
};
rep(i,n){
int v=a[i]+i;
int j=idsr[i];
psunr+=v;
pcntr+=1;
segr.set(j,segr.get(j)+v);
cntr.set(j,cntr.get(j)+1);
}
const int inf=1e18;
int ans=inf;
rep(i,n){
int l=max(i+1,n-i),r=1e9;
while(r-l>3){
int ml=(2*l+r)/3;
int mr=(l+2*r)/3;
if(af(i,ml)<af(i,mr)){
r=mr;
}else{
l=ml;
}
}
rng(j,l,r+1){
ans=min(ans,af(i,j));
}
int v=a[i]+i;
int j=idsr[i];
psunr-=v;
pcntr-=1;
segr.set(j,segr.get(j)-v);
cntr.set(j,cntr.get(j)-1);
v=a[i]-i;
j=idsl[i];
psunl+=v;
pcntl+=1;
segl.set(j,segl.get(j)+v);
cntl.set(j,cntl.get(j)+1);
}
print(ans);
}
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
10 ms |
468 KB |
Output is correct |
| 2 |
Correct |
14 ms |
468 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
10 ms |
436 KB |
Output is correct |
| 2 |
Correct |
12 ms |
468 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
18 ms |
596 KB |
Output is correct |
| 2 |
Correct |
19 ms |
596 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
26 ms |
980 KB |
Output is correct |
| 2 |
Correct |
39 ms |
676 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
27 ms |
724 KB |
Output is correct |
| 2 |
Correct |
32 ms |
728 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
68 ms |
1632 KB |
Output is correct |
| 2 |
Correct |
77 ms |
1084 KB |
Output is correct |
| 3 |
Correct |
42 ms |
980 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
443 ms |
5736 KB |
Output is correct |
| 2 |
Correct |
370 ms |
3608 KB |
Output is correct |
| 3 |
Correct |
486 ms |
5788 KB |
Output is correct |
| 4 |
Correct |
632 ms |
6052 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
833 ms |
6464 KB |
Output is correct |
| 2 |
Correct |
889 ms |
10892 KB |
Output is correct |
| 3 |
Correct |
810 ms |
10684 KB |
Output is correct |
| 4 |
Correct |
768 ms |
10616 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
1584 ms |
7620 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
1572 ms |
22116 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
