UVA11300 Spreading the Wealth

不见月光见星光

于 2021-08-18 15:22:45 发布

阅读量138

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/weixin_50187215/article/details/119781127

版权

UVA 算法线性代数 cpp

算法专栏收录该内容

30 篇文章 0 订阅

订阅专栏

UVA11300 Spreading the Wealth

请添加图片描述

思路

对于这道题，我们可以将其转化为一道线性代数的问题，设第 $i$ 个人的初始值为 $A_{i}$ ，第 $i$ 个人给第 $i - 1$ 个人的金币数目为 $x_{i}$ ，设最终值为 $a v g$ ，我们可以得到这样一个等式 $A_{1}-x_{1}+x_{2}=avg$ ，设 $C_{1} = A_{1}-avg$ 得到 $x_{1}-x_{2}=C_{1}\to x_{2}=x_{1}-C_{1}(1)$

根据这个式子我们对下面递推的式子进行处理 $A_{2}-x_{2}+x_{3}=a vg\to x_{2}-x_{3}=A_{2}-avg\to x_{3}=x_{2}-A_{2}+avg(2)$

将 $(1)$ 式带入 $(2)$ 式后，得到 $x_{3}=x_{1}-C_{2}(C_{2}=A_{1}+A_{2}-2avg)$

有这个式子我们可以对下面的n个式子进行递推，可以得到 $x_{n}=x_{1}-C_{n-1}$ ，而 $C_{n-1}$ 我们是可以解出来的，所以只剩下一个变量 $x_{1}$ ，这个式子就变成了一个单变量极值的题目，他的几何意义可以概括为:在一条坐标轴上，有n个坐标分别为 $C_{1},C_{2}...C_{n}$ 的点，我们要求这个数轴上一点到这个n个数距离之和的最小值，因为这个是一个环，所以最后一点得到的结果为 $x_{1}$

要求的结果就是 $min{|x_{1}|+|x_{1}-C_{1}|+....|x_{1}-C_{n-1}|}$ ，对于这个式子的解，我们可以得到当 $x_{1}$ 等于所有点的中位数的时候是最小的。证明这个结论只需要将这个点置于中点，然后向任意一边进行移动可以发现这个数值都是在变大

对于点个数为偶数的情况 $x_{1}$ 只需要在中间两个点的区间内即可

code

#include<iostream>
#include<algorithm>
#define N 1010000
#define ll long long
using namespace std;
ll A[N],C[N],sum,m;
int main()
{
    int n;
    while(scanf("%d",&n)==1)
    {
        sum = 1;
        for(int i = 1;i <= n;i++)
        {
            scanf("%lld",&A[i]);
            sum += A[i];
        }
        ll avg = sum/n;
        C[0] = 0;
        for(int i = 1;i < n;i++) C[i] = C[i-1] + A[i] - avg;
        sort(C,C+n);
        ll mid = C[n/2],ans = 0;
        for(int i = 0;i < n;i++) ans += abs(mid - C[i]);
        printf("%lld\n",ans);
    }
    return 0;
}