Submission #13211029

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
#define int ll

#define REP(i,n) for(int i=0;i<n;++i)
#define ALL(name) name.begin(),name.end()
#define SORT(name) sort(name.begin(), name.end())
#define ZERO(p) memset(p, 0, sizeof(p))
#define MINUS(p) memset(p, -1, sizeof(p))
#if 1
#  define DBG(fmt, ...) printf(fmt, ##__VA_ARGS__)
#else
#  define DBG(fmt, ...)
#endif

const ll LLINF = (1LL<<60);
const int INF = (1LL<<30);
const double DINF = std::numeric_limits<double>::infinity();
const int MOD = 1000000007;
#define MAX_N 100010

int N;
vector<int> A;

int func(int val) {
    int ret = 0;
    REP(i, N) {
        ret += abs(A[i] + val);
    }
    return ret;
}


// [lb, ub] の範囲の値を下に凸な func 関数で得られるとき、
// 三分探索で最小解を求める
// func は返り値 ll, 引数ひとつで型 ll の関数を与える
ll tri_search(ll lb, ll ub, function< ll(ll) > func) {
    while(ub - lb > 1) {
        ll t1 = (2 * lb + ub) / 3;
        ll t2 = (lb + 2 * ub) / 3;
        if(func(t1) > func(t2)) {   // 上に凸ならここの不等号を逆転させれば OK
            if(lb == t1) { break; } // もう変化しないくらい範囲が狭くなった
            lb = t1;
        }
        else {
            if(ub == t2) { break; }
            ub = t2;
        }
    }

    // この時点でだいたい ub - lb <= 6 くらいになってるはず
    ll ret = LLINF;                 // 上に凸ならここを -LLINF にする
    for(ll i = lb; i <= ub; ++i) {
        ret = min(ret, func(i));    // 上に凸ならここを max にする
    }
    return ret;
}

signed main() {
    cin.tie(0); cout.tie(0);
    ios::sync_with_stdio(false);

    cin >> N;
    A.resize(N);
    REP(i, N) { 
        cin >> A[i];
        A[i] -= (i + 1);
    }

    int ans = tri_search(-INF * 2, INF * 2, func);

    printf("%lld\n", ans);

    return 0;
}

Submission Info

Judge Result