[leetcode] 396. Rotate Function

我是链接

看到这题，是easy，应该没什么套路，感觉没什么特别的最优的性质，然后就暴力，每一个位置单独计算，一提交，tle，然后想想，easy的题目，应该没什么难的，仔细思考一下，然后肯定所有的情况都要遍历，然后关键就是怎么迅速的在每2个之间进行O(1)的转移，然后再画一画，恩，就是这样，然后就交了，就过了。感觉还是胆子大一些，大不了罚时多一些，罚时算是花时间买测试用例吧。其实仔细想想，之前的解法应该会超时，但是没数据范围，也无法大概的估算，所以不知道应该是否优化，咸鱼也是有梦想的，万一实现了呢！

class Solution {
public:
int work(vector<int> &a, int x) {
    int n = a.size();
    int res = 0;
    for (int i = 0; i < n; i++) {
        res += a[(x + i)% n] * i;
    }
    return res;
}
 int maxRotateFunction(vector<int>& A) {
        int n = A.size();
        if(n < 2) return 0;
        int res = work(A, 0), cur = res;
        int s = 0;
        for (int i = 0; i < n; i++) s += A[i];
        for (int i = 0; i < n; i++) {
            cur = cur + A[i] * (n - 1) - s + A[i];
            res = max(res, cur);
        }
        return res;
    }
};