LeetCode : Rotate Function

IMapMap

于 2016-09-25 16:03:17 发布

阅读量358

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/IMapMap/article/details/52662575

LeetCode 专栏收录该内容

25 篇文章

订阅专栏

本文介绍了一种针对数组旋转后元素与其下标乘积之和的最大值问题的高效求解算法。通过发现不同旋转状态之间的递推关系，避免了直接计算所带来的超时问题。文章给出了具体的实现代码，并通过加入预判条件进一步提升了算法效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先尝试直接计算，记录最大值，会超时。

思路

通过观察

$f(0) = 0 \times A[0] + 1 \times A[1] + 2 \times A[2] + 3 \times A[3].$
$f(1) = 1 \times A[0] + 2 \times A[1] + 3 \times A[2] + 0 \times A[3].$
$f(2) = 2 \times A[0] + 3 \times A[1] + 0 \times A[2] + 1 \times A[3].$
$f(3) = 3 \times A[0] + 0 \times A[1] + 1 \times A[2] + 2 \times A[3].$

发现function之间有递归关系

$f(1) = f(0) + sum - 4 \times A[3].$
$f(2) = f(1) + sum - 4 \times A[2].$
$f(3) = f(2) + sum - 4 \times A[1].$

Solution

int maxRotateFunction(vector<int>& A)
{
    int k = A.size();
    int sum = 0; // A中所有元素和
    int res = 0; // 结果：记录最大值
    int temp = 0; // function(i)

    for(int i = 0; i < k; i++)
    {
        sum += A[i];
        temp += i * A[i];
    }

    res = temp;

    for(int i = 1; i < k; i++)
    {
        temp = temp + sum - k * A[k - i % k];
        if(temp > res)
            res = temp;
    }

    return res;
}

Tip

前面加一些判断条件，明显减少了最后判定的时间。

if(k == 1)
    return 0;
else if(k == 2)
    return max(A[0], A[1]);