Educational Codeforces Round 98 (Rated for Div. 2) B. Toy Blocks

题意： 给定$n$个盒子，每个盒子有$a_i$个物品，问你至少需要往这些盒子中添加多少个物品，才能使得无论选择哪一个盒子$i$，将$a_i$分给剩余$n-1$个盒子，剩余$n-1$个盒子中的物品数都一样。
数据范围：$2\le n\le 10^5,0\le a_i\le 10^9$

题解：
考虑选择第$i$个盒子，剩下$n-1$个盒子中物品最多的盒子为$j$，分完第$i$个盒子后每个盒子物品数量为$ave$，那么必然有$ave\ge a[j]$。
故记录下前缀$max$和后缀$max$，然后遍历每个盒子即可。

代码：

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

typedef long long ll;

const int N = 1e5 + 10;
int a[N], n;
int pre[N], suf[N];
ll sum, res;

void solve() {
	scanf("%d", &n), sum = 0, res = 0;
	for(int i = 1; i <= n; ++i) scanf("%d", &a[i]), sum += a[i];
	if(sum % (n - 1)) res = n - 1 - sum % (n - 1);
	
	pre[0] = suf[n + 1] = 0;
	for(int i = 1; i <= n; ++i) pre[i] = max(pre[i - 1], a[i]);
	for(int i = n; i >= 1; --i) suf[i] = max(suf[i + 1], a[i]);
	
	for(int i = 1; i <= n; ++i) {
		ll mx = max(pre[i - 1], suf[i + 1]);
		ll ave = (res + sum) / (n - 1);
		if(ave < mx) res += (mx - ave) * (n - 1); 
	}
	printf("%lld\n", res);
}

int main()
{
	int T = 1;
	scanf("%d", &T);
	for(int i = 1; i <= T; ++i) solve();

	return 0;
}