题意
n n n 个人围成一桌,每个人都有 a i a_i ai 个金币,他们可以将一部分金币分给左手边的一个人和有手边的一个人。设每个人分给其他人的金币数量为 x i x_i xi,求所有的 x x x 加起来最小是多少。注意,每个人只能将金币给他相邻的人。题目保证金币总和能被 n n n 整除。
分析
设第
i
i
i个人给他左边伙计的金币数量为
x
i
x_i
xi,当
x
i
<
0
x_i<0
xi<0 表示其左边的人给他金币。那么有一个守恒关系:任何一个人在给出了或者得到金币以后,最后金币数都相等。
a
i
−
x
i
+
x
i
+
1
=
M
a_i-x_i+x_{i+1}=M
ai−xi+xi+1=M
有了这个以后我们可以得到下面的这个表格
1 1 1 | a 1 − x 1 + x 2 = M a_1-x_1+x_2=M a1−x1+x2=M |
---|---|
2 2 2 | a 2 − x 2 + x 3 = M a_2-x_2+x_3=M a2−x2+x3=M |
3 3 3 | a 3 − x 3 + x 4 = M a_3-x_3+x_4=M a3−x3+x4=M |
. . . ... ... | . . . ... ... |
n − 1 n-1 n−1 | a n − 1 − x n − 1 + x n = M a_{n-1}-x_{n-1}+x_n=M an−1−xn−1+xn=M |
n n n | a n − x n + x 1 = M a_n-x_n+x_1=M an−xn+x1=M |
下面我们做一个变形:从第一行开始,将当的等式加到下面一行。这样我们看第二个等式更新为了 a 1 + a 2 − x 1 + x 3 = 2 M a_1+a_2-x_1+x_3=2M a1+a2−x1+x3=2M,消去了 x 2 x_2 x2。然后再将这个更新以后的式子加到第三行。那么第三行就更新为 a 1 + a 2 + a 3 − x 1 + x 4 = 3 M a_1+a_2+a_3-x_1+x_4=3M a1+a2+a3−x1+x4=3M,消去了 x 3 x_3 x3。这样更新了所有的等式以后结果如下表。
1 1 1 | a 1 − x 1 + x 2 = M a_1-x_1+x_2=M a1−x1+x2=M |
---|---|
2 2 2 | a 1 + a 2 − x 1 + x 3 = 2 M a_1+a_2-x_1+x_3=2M a1+a2−x1+x3=2M |
3 3 3 | a 1 + a 2 + a 3 − x 1 + x 4 = 3 M a_1+a_2+a_3-x_1+x_4=3M a1+a2+a3−x1+x4=3M |
. . . ... ... | . . . ... ... |
n − 1 n-1 n−1 | ( a 1 + a 2 + . . . + a n − 1 ) − x 1 + x n = ( n − 1 ) M (a_1+a_2+...+a_{n-1})-x_{1}+x_n=(n-1)M (a1+a2+...+an−1)−x1+xn=(n−1)M |
n n n | ( a 1 + a 2 + . . . + a n ) − x 1 + x 1 = n M (a_1+a_2+...+a_n)-x_1+x_1=nM (a1+a2+...+an)−x1+x1=nM |
其中 a i a_i ai是我们已知的,所以全部的等式都变成了与 x 1 , x i x_1, x_i x1,xi 有关的了。显然,最后一个等式是一个恒等式,这里我们将其删除。然后将除 x 1 x_1 x1 的变量提到一边。
1 1 1 | x 2 = x 1 − ( a 1 − M ) x_2=x_1-(a_1-M) x2=x1−(a1−M) |
---|---|
2 2 2 | x 3 = x 1 − ( a 1 + a 2 − 2 M ) x_3=x_1-(a_1+a_2-2M) x3=x1−(a1+a2−2M) |
3 3 3 | x 4 = x 1 − ( a 1 + a 2 + a 3 − 3 M ) x_4=x_1-(a_1+a_2+a_3-3M) x4=x1−(a1+a2+a3−3M) |
. . . ... ... | . . . ... ... |
n − 1 n-1 n−1 | x n − 1 = x 1 − [ a 1 + a 2 + a 3 + a n − 1 − ( n − 1 ) M ] x_{n-1}=x_1-[a_1+a_2+a_3+a_{n-1}-(n-1)M] xn−1=x1−[a1+a2+a3+an−1−(n−1)M] |
如上表,我们最后的答案要求的是每个人转移的金币之和的最小值,也就是
∣
x
1
∣
+
∣
x
2
∣
+
.
.
.
+
∣
x
n
∣
|x_1|+|x_2|+...+|x_n|
∣x1∣+∣x2∣+...+∣xn∣ 的最小值。那么将上面的式子代代入进来我们可以看出就是求
数
轴
上
x
1
到
0
,
(
a
1
−
M
)
,
(
a
1
+
a
2
−
2
M
)
.
.
.
[
a
1
+
a
2
+
.
.
a
n
−
1
−
(
n
−
1
)
M
]
的
最
小
值
数轴上x1到0, (a1-M),(a1+a2-2M)...[a_1+a_2+..a_{n-1}-(n-1)M]的最小值
数轴上x1到0,(a1−M),(a1+a2−2M)...[a1+a2+..an−1−(n−1)M]的最小值
这个最小是应当是这些数的中位数。这个可以这么理解,假设这个点在
x
x
x,
x
x
x左边有a个点,右边有
b
b
b个点,且
a
<
b
a<b
a<b。那么如果x往右边偏移很小很小的一点,则左边的距离和会增加
a
d
ad
ad,右边的会减少
b
d
bd
bd,而
b
>
a
b>a
b>a,则肯定要往右边移动。那么我们可以判断出来,只有当
a
=
=
b
a==b
a==b的情况下才最小,如果总数是奇数个的话就是中位数,不然中间两个数之间的任意位置都能取最小。
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 1e6;
ll a[MAXN+5],c[MAXN+5];
int main()
{
int n;
while (scanf("%d",&n)!=EOF)
{
ll all = 0;
for (int i =0;i < n;i++)
{
scanf("%lld",&a[i]);
all += a[i];
}
ll m = all/n;
c[0] = 0;
for (int i = 1;i < n;i++)
{
c[i] = c[i-1] + m - a[i];
}
sort(c,c+n);
ll x = c[n/2];
ll ans = 0;
for (int i = 0;i<n;i++)
ans += abs(x-c[i]);
printf("%lld\n",ans);
}
return 0;
}