【每日一题】2681. 英雄的力量

2681. 英雄的力量

题目描述

给你一个下标从 0 开始的整数数组 nums ，它表示英雄的能力值。如果我们选出一部分英雄，这组英雄的 力量 定义为：

i0 ，i1 ，… ik 表示这组英雄在数组中的下标。那么这组英雄的力量为 max(nums[i0],nums[i1] … nums[ik])2 * min(nums[i0],nums[i1] … nums[ik]) 。
请你返回所有可能的 非空 英雄组的 力量 之和。由于答案可能非常大，请你将结果对 109 + 7 取余。

示例 1：

输入：nums = [2,1,4]
输出：141
解释：
第 1 组：[2] 的力量为 22 * 2 = 8 。
第 2 组：[1] 的力量为 12 * 1 = 1 。
第 3 组：[4] 的力量为 42 * 4 = 64 。
第 4 组：[2,1] 的力量为 22 * 1 = 4 。
第 5 组：[2,4] 的力量为 42 * 2 = 32 。
第 6 组：[1,4] 的力量为 42 * 1 = 16 。
第​ ​​​​​​7 组：[2,1,4] 的力量为 42​​​​​​​ * 1 = 16 。
所有英雄组的力量之和为 8 + 1 + 64 + 4 + 32 + 16 + 16 = 141 。

示例 2：

输入：nums = [1,1,1]
输出：7
解释：总共有 7 个英雄组，每一组的力量都是 1 。所以所有英雄组的力量之和为 7 。

提示：

1 <= nums.length <= 105
1 <= nums[i] <= 109

解题思路

思路：注意，该题是选出一部分数组，但是不要求是连续数组，又涉及到数组最大值和最小值，故可以对该数组进行排序。假设一个数组排序后得到a、b、c、d、e，我们依次枚举最大值：假设当前最大值为a，那么得到的力量为a²*a；假设当前最大值为b，那么得到的力量为b²*(a*2⁰+b)；假设当前最大值为c，那么得到的力量为c²*(a*2¹+b*2⁰+c)；假设当前最大值为d，那么得到的力量为d²*(a*2²+b*2¹+c*2⁰+d)；假设当前最大值为e，那么得到的力量为e²*(a*2³+b*2²+c*2¹+d*2⁰+e)；如此我们可以得到递推公式，即假设递推元素s=0，当前元素为x，res+=x²*(s+x)，下一轮递推元素为s’=2*s+x。注意MOD运算公式，(a+b)%MOD=a%MOD+b%MOD，(a*b)%MOD=a%MOD*b%MOD。

class Solution {
public:
    int sumOfPower(vector<int>& nums) {
        // 取模
        const int MOD=1e9+7;
        // 排序
        sort(nums.begin(),nums.end());
        int res=0,s=0;
        // 遍历 当前x作为最大值
        for(long long x:nums)
        {
            res=(res%MOD+x%MOD*x%MOD*(x+s)%MOD)%MOD;
            s=(2*s+x)%MOD;
        }
        return res;
    }
};

总结：该题为子数组，但不要求是连续子数组，故相对处理简单。