The 2021 ICPC Asia Jinan Regional Contest D Arithmetic Sequence（三分+货舱选址）

最新推荐文章于 2022-09-04 20:47:39 发布

JXNU_SONG

最新推荐文章于 2022-09-04 20:47:39 发布

阅读量675

点赞数 1

分类专栏：思维三分文章标签：算法

本文链接：https://blog.csdn.net/qq_45863710/article/details/121380266

版权

思维同时被 2 个专栏收录

55 篇文章 1 订阅

订阅专栏

三分

1 篇文章 0 订阅

订阅专栏

Arithmetic Sequence

在这里插入图片描述
题目大意：给出一个序列 $a$ ，一步操作可以将一个数加一或减一，问将整个序列变成等差数列的最小操作数。

我们发现，如果公差确定，最小值是可以 $O (n)$ 求得的。我们设公差为已知量 $x$ ，首项为未知量 $c_1$ 。那么那么整体的代价就是
$\sum\limits_{i=1}^n|a_i-(c_1+(i-1)×x)|$
其实这里已经有点像货舱选址的问题，但是，我们没有一个确定的终点，那我们可以想如何将终点变为确定。那么我们可以对于 $i$ 号点来说同时下移 $(i - 1) \times x$ ，形式化的说如下：
$\sum\limits_{i=1}^n|(a_i-(i-1)×x)-c_1|$

我们设 $t_i=a_i-(i-1)×x$ ，则有：
$\sum\limits_{i=1}^n|t_i-c_1|$

那么这就变成了经典的货舱选址问题，也就是要使上述式子的值最小，首项 $c_1$ 应该取 $t_i$ 的中位数，求中位数可以直接用 $nth\_element$ 函数比较快的得到。

那么剩下就要确定最优公差了。那么我们假设公差为 $x$ ，另序列变为等差的最小代价为 $f (x)$ ，我们会发现这是一个凹函数。证明如题解中的说明：
在这里插入图片描述
最终我们就是三分公差，求得在此公差下的最小值，最终即可得到答案。

注意这题的数据范围可能会爆 $l o n g l o n g$ ，所以用 $_\_int128$ 比较稳妥。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 2e5 + 10;

int n, idx;

__int128 a[N], t[N];

inline __int128 read(){
    __int128 x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}

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

__int128 f(__int128 x) {
	for (int i = 1; i <= n; ++i) {
		t[i] = a[i] - 1ll * (i - 1) * x;
	}
	nth_element(t + 1, t + idx, t + 1 + n);
	__int128 cur = t[idx], cost = 0;
	for (int i = 1; i <= n; ++i) {
		// cost += abs(a[i] - cur);
		if (a[i] > cur) cost += a[i] - cur;
		else cost += cur - a[i];
		cur += x;
	}
	return cost;
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	scanf("%d", &n); idx = (n + 1) / 2;
	for (int i = 1; i <= n; ++i) {
		a[i] = read();
	}
	__int128 l = -1e13, r = 1e13;
	while(l + 3 < r) {
		__int128 lmid = l + (r - l) / 3;
		__int128 rmid = r - (r - l) / 3;
		__int128 lans = f(lmid);
		__int128 rans = f(rmid);
		if (lans > rans) l = lmid + 1;
		else r = rmid - 1;
	}
	__int128 ans = 1e25;
	for (__int128 i = l; i <= r; ++i) {
		ans = min(ans, f(i));
	}
	print(ans); puts("");
}