110. 数蝌蚪

单点时限: 2.0 sec

内存限制: 256 MB

有 n 个装着小蝌蚪的水缸排成一排，你拥有一个无限蝌蚪的袋子，可以往一个水缸里放入一只蝌蚪，也可以取出一只蝌蚪，求最少的操作数，使得每个水缸的蝌蚪数量形成一个公差为 k 等差数列。

输入格式
第一行一个数 n,k(3⩽n⩽3×105,0⩽k⩽104)。
第二行 n 个数，表示每个水缸里的蝌蚪数目(0⩽ai⩽104)。

输出格式
输出最少操作次数。

样例
input
4 2
1 2 3 4
output
4
input
4 2
0 1 2 3
output
6
提示
蝌蚪的个数不能是负的。

/*
思路一：拟合一条y=kx+b直线，使其与各点数差值之和最小。但会超时。
*/
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
	long long n,k;
	cin>>n>>k;
	int a[n+1];
	int m;
	for(int i = 1; i <= n; i++)
	{
		cin>>a[i];;
		m=max(m,a[i]);
	}
	long long sum=1e18;;
	for(int b=-k;b<=20;b++)
	{
		long long sum1=0;
		for(int i=1; i<=n; i++)
		{
			
			sum1+=abs(i*k+b-a[i]);
		} 
		sum=min(sum1,sum);
	}
	cout<<sum;
}

/*
思路二：转换y=kx+b为拟合y=b,将各点均先减去k*x；剪完之后的各点排序，可以证明b为中间一个数时差值最小。
*/
#include<iostream>
#include<algorithm>
using namespace std;
int main() {
	long long n,k;
	cin>>n>>k;
	long long a[n];
	long long sum=0,m;
	for(int i = 0; i < n; i++) {
		cin>>m;
		a[i]=m-i*k;
	}
	sort(a,a+n);
	long long b;
	if(n&1)//奇数时就是中间
		b=a[n/2];
	else//偶数时要为中间数的均值
		b=(a[n/2]+a[n/2-1])/2;
	if(b<0)//首项不能小于0
	    b=0;
	for(int i =0; i < n; i++) {
		sum+=abs(a[i]-b);
	}
	cout<<sum<<endl;
	return 0;
}