oj.uz

답안 #858914

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
858914 2023-10-09T11:48:21 Z Ahmed57 Krov (COCI17_krov) C++17
126 / 140
 559 ms 47288 KB 
#include <bits/stdc++.h>
using namespace std;
int cn = 0;
#define int long long
struct lol{
long long sum[600001] = {0},cnt[600001] = {0};
void addSum(int e,int v){
    while(e<=cn){
        sum[e]+=v;
        e+=e&-e;
    }
}void addcnt(int e,int v){
    while(e<=cn){
        cnt[e]+=v;
        e+=e&-e;
    }
}long long querysum(int e){
    long long res = 0;
    while(e>=1){
        res+=sum[e];
        e-=e&-e;
    }
    return res;
}long long querycnt(int e){
    long long res = 0;
    while(e>=1){
        res+=cnt[e];
        e-=e&-e;
    }
    return res;
}
}pref,suf;
signed main(){
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int n;
    cin>>n;
    long long arr[n];
    for(int i = 0;i<n;i++){
        cin>>arr[i];
    }
    long long rev[n*3+1] = {0};
    map<int,int> sav;set<int> comp;
    vector<int> re,dr;
    for(int i = 0;i<n;i++){comp.insert(arr[i]-i);comp.insert(arr[i]+i);re.push_back(arr[i]+i);dr.push_back(arr[i]-i);dr.push_back(max(i+1,(n-i))-i);comp.insert(max(i+1,(n-i))-i);}
    for(auto i:comp){
        sav[i] = ++cn;
        rev[cn] = i;
    }
    long long val[n] = {0};
    for(int i = 0;i<n;i++){
        suf.addSum(sav[arr[i]+i],arr[i]+i);
        suf.addcnt(sav[arr[i]+i],1);
    }
    sort(re.begin(),re.end());
    sort(dr.begin(),dr.end());
    long long mi = 2e18;
    for(int i = 0;i<n;i++){
        suf.addSum(sav[arr[i]+i],-(arr[i]+i));
        suf.addcnt(sav[arr[i]+i],-1);
        pref.addSum(sav[arr[i]-i],arr[i]-i);
        pref.addcnt(sav[arr[i]-i],1);
        int ze = max(i+1,(n-i))-i;
        int l = 1, r = cn , ans = 1e9;
        l = lower_bound(dr.begin(),dr.end(),ze)-dr.begin();
        l = sav[dr[l]];
        while(l<=r){
            int mid = (l+r)/2;
            int it = upper_bound(re.begin(),re.end(),rev[mid]+2*i)-re.begin()-1;

            if(pref.querycnt(mid)+(it>=0?suf.querycnt(sav[re[it]]):0)>=(n+1)/2){
                ans = mid;
                r = mid-1;
            }else l = mid+1;
        }
        //cout<<rev[ans]<<" "<<pref.querycnt(ans)+suf.querycnt(ans+2*i)<<endl;
        long long CNT = pref.querycnt(ans) , SUM = pref.querysum(ans);
        long long lo = 0;
        lo+=(rev[ans])*CNT-SUM;
        CNT = pref.querycnt(cn)-CNT;
        SUM = pref.querysum(cn)-SUM;
        lo+=SUM-rev[ans]*CNT;
        //cout<<lo<<endl;
        int it = lower_bound(re.begin(),re.end(),rev[ans]+2*i+1)-re.begin()-1;
        int de = (it>=0?sav[re[it]]:0);
        CNT = suf.querycnt(de) , SUM = suf.querysum(de);
        lo+=(rev[ans]+2*i)*CNT-SUM;
        CNT = suf.querycnt(cn)-CNT;
        SUM = suf.querysum(cn)-SUM;
        lo+=SUM-(rev[ans]+2*i)*CNT;
        //cout<<lo<<endl;
        mi = min(mi,lo);
    }
    cout<<mi<<endl;
}
/*
4 5 7 2 2
pref : 4 4 5 -1 -2
suf : 4 6 9 5 6
rev[ans]+2*i
1: 5
2 , 3 , 4 , 5 : 3
*/

Compilation message

krov.cpp: In function 'int main()':
krov.cpp:49:15: warning: unused variable 'val' [-Wunused-variable]
   49 |     long long val[n] = {0};
      |               ^~~

Subtask #1
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 6748 KB Output is correct
2 Correct 3 ms 6748 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 6748 KB Output is correct
2 Correct 3 ms 6748 KB Output is correct

Subtask #3
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 7000 KB Output is correct
2 Correct 5 ms 7004 KB Output is correct

Subtask #4
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 7260 KB Output is correct
2 Correct 5 ms 7140 KB Output is correct

Subtask #5
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 7256 KB Output is correct
2 Correct 7 ms 7260 KB Output is correct

Subtask #6
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 14 ms 8284 KB Output is correct
2 Correct 12 ms 7516 KB Output is correct
3 Correct 7 ms 7516 KB Output is correct

Subtask #7
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 103 ms 15932 KB Output is correct
2 Correct 93 ms 14700 KB Output is correct
3 Correct 109 ms 15820 KB Output is correct
4 Correct 141 ms 17520 KB Output is correct

Subtask #8
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 177 ms 19904 KB Output is correct
2 Correct 208 ms 21700 KB Output is correct
3 Correct 158 ms 23724 KB Output is correct
4 Correct 137 ms 23304 KB Output is correct

Subtask #9
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 273 ms 23596 KB Output is correct
2 Correct 292 ms 36324 KB Output is correct
3 Correct 158 ms 19100 KB Output is correct
4 Correct 359 ms 31668 KB Output is correct

Subtask #10
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 559 ms 47288 KB Output isn't correct
2 Halted 0 ms 0 KB -