(nice!!!)LeetCode 2681. 英雄的力量（数学、排序）

2681. 英雄的力量

思路：求的是力量之和，力量之和只有最大值和最小值有关，也就是和顺序无关。那么先将数组nums进行一个升序排列，枚举最小值和最大值下标i,j，然后再再枚举中间这几个数取还是不取，即2^(j-i-1)。
排完后，一层for循环开始枚举最大值，然后最小值也需要进行枚举。如果用for循环枚举最小值，那么时间复杂度0(n^2)，会超时。这里观察规律会发现，当前最大值和所有最小值的乘积，在轮到下一个最大值时，和这些最小值的乘积会翻一倍，因为他们中间的距离多了一个元素。利用这个规律，可以做到0(1)实现最小值的枚举。最后时间复杂度为0(nlogn+n)

图片来源

class Solution {
public:
    int sumOfPower(vector<int>& nums) {
        int n=nums.size();
        //升序排序
        sort(nums.begin(),nums.end());
        int mod=1e9+7;
        //记录力量之和
        long long ans=0;
        //记录最小值的枚举
        long long sum=0;
        for(int i=0;i<n;i++){
            ans=(ans+(long long)nums[i]*nums[i]%mod*(sum+nums[i])%mod)%mod;
            sum=(sum*2+nums[i])%mod;
        }
        return ans%mod;
    }
};