LeetCode 396. 旋转函数

最新推荐文章于 2024-07-24 14:54:00 发布

小明不要写Bug

最新推荐文章于 2024-07-24 14:54:00 发布

阅读量118

点赞数

分类专栏： LeetCode 每日一题题解文章标签： leetcode c++ java

本文链接：https://blog.csdn.net/qq_46176960/article/details/124338199

版权

LeetCode 每日一题题解专栏收录该内容

92 篇文章 2 订阅

订阅专栏

396. 旋转函数

题目来源：396. 旋转函数

2022.04.22 每日一题

LeetCode 题解持续更新中GitHub仓库地址 CSDN博客地址

今天的题目可以使用迭代的方法来完成，我们根据题目可以知道 F(k) 的计算方法，我们要求解的是 F(k) 的最大值，如果要暴力求解，从头计算出每一个的值得话，可能会导致计算量过大，导致时间超时。

那么我们有没有什么方法能够计算出每一个的值，然后寻求出最大值呢？

我们可以来找一找规律。

我们根据题目给的样例可以发现：

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 (1) = F (0) + c n t - l e n * n u m s [3]$

其中 cnt 代表数组中各个元素相加之和，len 代表数组的长度。

我们可以发现存在这样的关系 $F (i + 1) = F (i) + c n t + l e n * n u m s [l e n - i - 1]$ 的关系。

同时，我们会发现，如果数组的长度小于 2 的话，根据题意，其可以得到的最大值就是 0，直接返回即可。

有了规律以后，我们就可以来进行结果的迭代，取得最终的结果。

class Solution {
public:
    int maxRotateFunction(vector<int> &nums) {
        // 统计 字符串长度
        int len = nums.size();
        // 如果长度小于 2，可以直接返回 0
        if (len < 2) return 0;
        // 创建两个变量，一个统计当前的最大值，
        // 一个统计数组加和
        int res = 0, cnt = 0;
        // 求数组之和
        for (int num : nums) cnt += num;
        // 求 F(0)
        for (int i = 0; i < len; i++) res += i * nums[i];
        int cur=res;
        // 依次求出每一个值，然后进行比较
        for (int i = len - 1; i >= 0; i--) res = max(res, cur += cnt - len * nums[i]);
        return res;
    }
};

class Solution {
    public int maxRotateFunction(int[] nums) {
        // 统计 字符串长度
        int len = nums.length;
        // 如果长度小于 2，可以直接返回 0
        if (len < 2) return 0;
        // 创建两个变量，一个统计当前的最大值，
        // 一个统计数组加和
        int res = 0, cnt = 0;
        // 求数组之和
        for (int num : nums) cnt += num;
        // 求 F(0)
        for (int i = 0; i < len; i++) res += i * nums[i];
        int cur=res;
        // 依次求出每一个值，然后进行比较
        for(int i=len-1;i>=0;i--) res=Math.max(res, cur+=cnt-len*nums[i]);
        // 返回最大值
        return res;
    }
}