leetcode刷题(2022.05.09)

leetcode刷题(2022.05.09)

今天中午写的题。本来不准备写文章了，想一想还是坚持一下。最近事情有点多了。这道题有去看题解。

2171.Removing Minimum Number of Magic Beans

难度：中等

题目大意

给一个数组，可以任意减少每一个位置的数字，最后数组所有的数字都为0或者同一个数字。返回减少的值，要求减少值最小。

解题思路

 当然最开始想着直接暴力。然后写了两重循环交。果不其然直接超时。。。
 然后开始改进，想着先排个序然后两重循环里面加个判断相同的数值对应的结果相同。然后交。还是超限(毕竟$n^2$的本质没有改变)。
 然后观察，排序之后每一个位置上的结果不一定要通过遍历来求，只需要预先知道数组的总和sum，然后设数组长度为n，
那么当前位置上的结果:
$$ans=sum-beans[i]\ast (n-i)$$
因为数组是有序的，在当前位置之前的元素都可以视为小于当前元素，在之后的都是大于当前元素，在之前的都会置为0，在之后的都置为当前元素即$beans[i]\ast (n-i)$那么要减去的大小就是使用sum减去剩余的即可，同时每次都要最小值。

代码

class Solution {
public:
    long long minimumRemoval(vector<int>& beans) {
        long long temp=0,sum=0;
        sort(beans.begin(),beans.end());
        sum=accumulate(beans.begin(),beans.end(),0LL);
        long long ans=sum;
        for(int i=0;i<beans.size();++i)
        {
            temp=sum-(beans.size()-i)*beans[i];
            ans=min(ans,temp);
        }
        return ans;
    }
};

过题情况

优化

没有继续优化。看了题解区还是有很多优化方法的。不过这次偷个懒。。
这次学到了一个accumulate（start,last,num）这个函数，用于计算总和，num是初始大小就是在这个基础上加。