LeetCode刷题笔记-396.旋转函数

C语言

#define MAX_ONE(a,b) ((a)>(b)?(a):(b))
#define INT_MIN 0x80000000
int maxRotateFunction(int* nums, int numsSize){
    int dp = 0;
    int max = INT_MIN;
    int sum = 0;
    int k;
    int i;

    for (i = 0; i < numsSize; i++) {
        dp+=(i * nums[i]);
        sum += nums[i];
    }
    max = MAX_ONE(max, dp);
    for (k = 1; k < numsSize; k++) {
        dp = dp + sum - numsSize * nums[numsSize - k];
        max = MAX_ONE(max, dp);
    }
    return max;
}

注意点

1. 数学归纳找出F(k)与F(k-1)的关系为 F(k)=F(k-1)+sum(nums[:])-numsSize*nums[numsSize-k]
2. 注意max的初始化应该为INT_MIN即0x80000000来满足负数的情况;
3. 重复计算sum会导致超时，所以需要抽出来只做一次的运算；
4. dp代替dp[]优化空间；

结果

题目