最少移动

$$\mathrm{最少移动}$$

题目链接：$\mathrm{n}\mathrm{o}\mathrm{w}\mathrm{c}\mathrm{o}\mathrm{d}\mathrm{e}\mathrm{r}213315$

关于这场比赛

——>点我可以查看其它题目（目录）<——

到牛客看：

——>点我跳转<——

题目

给一个长度为 $n$ 的正整数序列 { a 1 , a 2 , . . . , a n } \{a_1,a_2,...,a_n\} {a1​,a2​,...,an​}，每次操作可以选择两个相邻的位置，让一个元素 $-1$ 另一个元素 $+1$，输出最少几次操作，能让所有元素相等，如果不可能实现，请输出 “-1”，不含引号。

输入

第一行一个整数 $T(T\le 20)$，表示 $T$ 组数据。

每组数据第一行一个整数 $n$，第二行 $n$ 个数字表示 $a$ 序列，$1\le a[i]\le 100000$

输出

对于每组数据，输出一个整数表示答案。

样例输入

3
3
1 3 2
3
2 2 3
5
1 2 3 1 3

样例输出

1
-1
3

数据范围

保证 $T\le 20$

• $10\%$ 的测试数据，$1\le n\le 5$
• $20\%$ 的测试数据，$1\le n\le 100$
• $50\%$ 的测试数据，$1\le n\le 10^3$
• $80\%$ 的测试数据，$1\le n\le 10^4$
• $100\%$ 的测试数据，$1\le n\le 10^5$

思路

这道题我觉得挺模拟的。

我们可以发现，操作让一个数 $+1$，另一个数 $-1$。那这些数的总和是一直不变的。
那我们又要让它们变成全部相同的数，容易想到是变成所有数的平均数。
也可以得知，如果平均数是小数，那就没有解。
（因为每次只能 $\pm 1$，那无论怎么搞都是整数）

那又因为每次只会改变两个相邻的数，那我们直接从一开始枚举，然后把第 $i$ 个直接变成平均数，然后造成的结果全部怼给 $i+1$。
然后一直下去，记录每次把每个数变成平均数要多少步，加在一起，就是答案了。

注意！要开 long long！！！

比赛时

看了一会想到了要平均数然后每一个数到平均数。

结果忘了要开 long long，没了 $20$ 分，血亏QAQ

代码

#include<cstdio>
#include<algorithm>
#define ll long long 

using namespace std;

ll T, n, a[100001], sum, mid, ans, add;
ll re, zhengfu;
char c;

ll read() {
	re = 0;
	zhengfu = 1;
	c = getchar();
	
	while (c < '0' || c > '9') {
		c = getchar();
	}
	while (c >= '0' && c <= '9') {
		re = re * 10 + c - '0';
		c = getchar();
	}
	
	return re * zhengfu;
}

int main() {
	T = read();
	for (int times = 1; times <= T; times++) {
		sum = ans = 0;
		
		n = read();
		for (int i = 1; i <= n; i++) {
			a[i] = read();
			sum += a[i];
		}
		
		if (sum % n != 0) {//不能每个数都到平均数，就是不能每个数到同一个数
			printf("-1\n");
			continue;
		}
		mid = sum / n;//算出平均数
		
		for (int i = 1; i < n; i++) {
			add = a[i] - mid;
			a[i + 1] += add;//这个数走到平均数对下一个数的改变
			ans += abs(add);//每个数到平均数要多少步
		}
		
		printf("%lld\n", ans);
	}
	
	return 0;
}