2023-08-01 LeetCode每日一题（英雄的力量）-Toy模板网

这篇具有很好参考价值的文章主要介绍了2023-08-01 LeetCode每日一题（英雄的力量）。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

2023-08-01每日一题

一、题目编号

2681. 英雄的力量

二、题目链接

点击跳转到题目位置

三、题目描述

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

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

示例1：
2023-08-01 LeetCode每日一题（英雄的力量）,LeetCode每日一题,leetcode,算法,数据结构
示例2：

提示：

1 <= nums.length <= 10⁵
1 <= nums[i] <= 10⁹

四、解题代码

class Solution {
public:
    int sumOfPower(vector<int>& nums) {
        int n = nums.size();
        sort(nums.begin(), nums.end());
        vector<int> dp(n);
        vector<int> preSum(n + 1);
        int res = 0, mod = 1e9 + 7;
        for (int i = 0; i < n; i++) {
            dp[i] = (nums[i] + preSum[i]) % mod;
            preSum[i + 1] = (preSum[i] + dp[i]) % mod;
            res = (int) ((res + (long long) nums[i] * nums[i] % mod * dp[i]) % mod);
            if (res < 0) {
                res += mod;
            }
        }
        return res;
    }
};