用 Python 编写程序，用于查找旋转数组的最大加权和

pythonserver side programmingprogramming更新于 2025/12/7 14:04:17

假设我们有一个包含几个元素的数组。如果数组元素被旋转，我们必须找到最大加权和。数组 nums 的加权和可以像下面这样计算 −

$$\mathrm{𝑆=\sum_{\substack{𝑖=1}}^{n}𝑖∗𝑛𝑢𝑚𝑠[𝑖]}$$

因此，如果输入为 L = [5,3,4]，则输出将为 26，因为

数组为 [5,3,4]，和为 5 + 2*3 + 3*4 = 5 + 6 + 12 = 23
数组为 [3,4,5]，和为 3 + 2*4 + 3*5 = 3 + 8 + 15 = 26(最大值)
数组为[4,5,3]，总和为 4 + 2*5 + 3*3 = 4 + 10 + 9 = 23

为了解决这个问题，我们将遵循以下步骤 −

n := nums 的大小
sum_a := nums 中所有元素的总和
ans := (nums[i] *(i + 1)) 中所有元素的总和，对于 0 到 n 范围内的所有 i
cur_val := ans
对于 0 到 n - 1 范围内的 i，执行
- cur_val := cur_val - sum_a + nums[i] * n
- ans := ans 和 cur_val 的最大值
返回 ans

示例

让我们看看下面的实现以便更好地理解 −

def solve(nums):
    n = len(nums)
    sum_a = sum(nums)
    cur_val = ans = sum(nums[i] * (i + 1) for i in range(n))
   
    for i in range(n):
        cur_val = cur_val - sum_a + nums[i] * n
        ans = max(ans, cur_val)
   
    return ans

nums = [5,3,4]
print(solve(nums))