2021济南 D Arithmetic Sequence （三分）（银牌题）

PTA的教育超市可以找到这场比赛的题。

这题属于银牌题的级别，该题有一个应该算是常识性的知识点，即在一个序列中，sum(|a[i] - x|)要取得最小值，x应该为这个序列中的中位数，即第(n + 1) / 2 大的数。然后利用这一性质可以三分求答案。

很奇怪的一点是答案要用__int128来保存，否则存储不下这么大的数。有人解释说，在计算等差数列时，中位数可能达到1e18的级别，所以long long不够用。

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5 + 5;
using ll = long long;

int n;
ll a[N], c[N];

void print(__int128 x) {
	if (x > 9)
		print(x / 10);
	putchar(x % 10 + '0');
}

__int128 cal(ll d) {
	for (int i = 1; i <= n; i++)
		c[i] = a[i] - d * (i - 1);
	nth_element(c + 1, c + n / 2 + 1, c + 1 + n);
	__int128 res = 0;
	ll x = c[n / 2 + 1];
	for (int i = 1; i <= n; i++) {
		__int128 y = x - c[i];
		if (y > 0) res += y;
		else res -= y;
	}
	return res;
}

int main(void) {
//	freopen("in.txt", "r", stdin);
	scanf("%d", &n);
	for (int i = 1; i <= n; i++)
		scanf("%lld", &a[i]);
	ll l = -1e13, r = 1e13;
	while (l < r) {
		ll d1 = l + (r - l) / 3;
		ll d2 = r - (r - l) / 3;
		__int128 c1 = cal(d1);
		__int128 c2 = cal(d2);
		if (c1 > c2) l = d1 + 1;
		else r = d2 - 1;
	}
	__int128 ans = min(cal(r), cal(r + 1));
	print(ans);
	return 0;
}