每日一题(2681. 英雄的力量)-动态规划/前缀和

题目

2681. 英雄的力量

题解思路

nums数组代表英雄的能力值

需要计算下标从i…j范围内的英雄能力值，计算过程如下

【nums1,nums2,nums3,nums4.....nums[k] 】 对于从1到k这个非空的数组，其结果值为nums[k] * min(nums1, nums2, nums3, nums4.....nums[k])^2 

---

既然需要计算所有的非空数组，我们将数组从小到大排序之后：

• 对于数组中任意一个元素来说，以其为下标结尾的的子数组，最大的元素就是该元素

同时我们利用动态数组dp[i]表示以i为结点的最小值之和。

1. 对于以1…i为下标的子数组的最小值之和为dp[i]

2. 同时利用前缀和pre来记录最小值之和

3. 最后针对以i为结尾的数组计算 ，可以简化为如下

   res = dp[i] * (nums[i] * nums[i])
   

代码

C++

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

Python

class Solution:
    def sumOfPower(self, nums: List[int]) -> int:
        nums.sort()
        dp = [0 for _ in range(len(nums))]
        pre_sum = [0 for _ in range(len(nums) + 1)]
        res = 0
        mod = 10 ** 9 + 7
        for i in range(len(nums)):
            dp[i] = (nums[i] + pre_sum[i]) % mod
            pre_sum[i + 1] = (pre_sum[i] + dp[i]) % mod
            res = (res + nums[i] * nums[i] * dp[i]) % mod 
        return res