leetcode(102)_396_medium_旋转函数_python

旋转函数

题目描述：
给定一个长度为 n 的整数数组 A 。
假设 Bk 是数组 A 顺时针旋转 k 个位置后的数组，我们定义 A 的“旋转函数” F 为：
F(k) = 0 * Bk[0] + 1 * Bk[1] + … + (n-1) * Bk[n-1]。
计算F(0), F(1), …, F(n-1)中的最大值。
注意:
可以认为 n 的值小于 105。
示例:
A = [4, 3, 2, 6]
F(0) = (0 * 4) + (1 * 3) + (2 * 2) + (3 * 6) = 0 + 3 + 4 + 18 = 25
F(1) = (0 * 6) + (1 * 4) + (2 * 3) + (3 * 2) = 0 + 4 + 6 + 6 = 16
F(2) = (0 * 2) + (1 * 6) + (2 * 4) + (3 * 3) = 0 + 6 + 8 + 9 = 23
F(3) = (0 * 3) + (1 * 2) + (2 * 6) + (3 * 4) = 0 + 2 + 12 + 12 = 26
所以 F(0), F(1), F(2), F(3) 中的最大值是 F(3) = 26 。
提示：

• 2 <= nums.length <= 10⁴
• -10⁹ <= nums[i] <= 10⁹
• -10⁹ <= target <= 10⁹
• 只会存在一个有效答案

解法

考虑向右移动一次，我们先假设最后一个元素不会回到首位置，而是后移一个位置，那么新的函数值与旧的函数值的差值就是整个数组元素的加和。而实际上，末位元素要回到首位置，因此新的函数值还应该减去 len(nums)*ele，ele 是末位元素的值。因此我们只需要逐个更新即可，然后找出最大值。

代码

class Solution:
    def maxRotateFunction(self, nums: List[int]) -> int:
        sum_nums = sum(nums)
        sum_candidate = reduce(lambda x, y: x + y, [num * idx for idx, num in enumerate(nums)])
        res = sum_candidate
        for i in range(len(nums) - 1, -1, -1):
            sum_candidate = sum_candidate + sum_nums - nums[i] * len(nums)
            res = max(res, sum_candidate)
        return res
      

测试结果

执行用时：364 ms, 在所有 Python3 提交中击败了 21.90% 的用户
内存消耗：24.1 MB, 在所有 Python3 提交中击败了 5.24% 的用户

说明

算法题来源：力扣(LeetCode)