[蓝桥杯2018初赛]付账问题

贪心算法

如输入用例：

平均每个人要付3元，那么我们可以看到有3个人，它们带的钱是小于等于平均数的，那么这些人应该把自己的钱全付出去，因为全付出去的方差是最小的。然后我们就得到了一个新的子问题，那就是剩下7个人付24元，怎么付离平均值3的标准差最小。先说为什么这个子问题的最优解可以得出最终问题的最优解，我们知道这个问题满足最优子结构性质——也就是构成这个问题最优解的子问题的解也是最优的，可以用反证法很容易证明。反过来说是不真的，也就是说不是每个子问题的最优解都是最终问题最优解的一部分，我们必须要比较每一个于最终的问题有关的子问题的最优解，这就是动态规划的思想，比如01背包问题，与最终问题有关的两个子问题是1.拿了第n件物品后用剩下的容量去拿其他n-1件物品所能得到的最大价值和2.不拿第n件物品以原始的容量去拿其他n-1件物品所能得到的最大价值，两个子问题分别对应着两种情况，因为我们无法判断哪一种情况是最优的，所以我们需要比较二者。但是这题因为有贪心最优性质，所以也就排除了其他的情况可以得到最优解的可能，所以这个子问题的最优解就是最终问题最优解的一部分。那我们怎么求得这个子问题的最优解呢？如果当他们中有人带的钱比他们所需付的平均值24/7=3.42857142857142小的时候，我们可以用反证法再次证明他们应该把自己的钱全付出去，具体来说就是 （a-0.2）^2 + (b+0.2)^2比a^2 + b^2要大 ，从而再次缩小子问题。直到每个人带的钱都比他们应该付的平均值要多，这个时候可以证明每个人只需要付他们此时应该付的平均值就可以离他们最开始应该付的平均值的方差是最小的，还是用反证法来证明。

代码如下，不知道为什么只能过一半的用例，还有50%时间超限：

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
    public static void main(String[] args){
        Scanner scanner=new Scanner(System.in);
        int n=scanner.nextInt();
        long S=scanner.nextLong();
        ArrayList<Long> data=new ArrayList<>();
        for (int i = 0; i < n; i++) {
            long temp=scanner.nextLong();
            data.add(temp);
        }
        double aver=S/n,newAver=aver,res=0;
        while (!data.isEmpty()){
            int bigger=data.size();
            for (int i=0;i<data.size();i++){
                if (data.get(i)<=newAver){
                    res+=Math.pow(Math.abs(data.get(i)-aver),2);
                    S-=data.get(i);
                    data.remove(i);
                    i--;
                }
            }
            if (bigger!=data.size()){
                newAver=((double)S)/data.size();
            }
            else {
                for (Long temp:data){
                    res+=Math.pow(newAver-aver,2);
                }
                data.clear();
            }
        }
        System.out.printf("%.4f",Math.sqrt(res/n));
    }

}