Rotate Function Solutions in C++

// Source : https://leetcode.com/problems/rotate-function/
// Author : Hao Chen
// Date   : 2016-11-03




// 
// Asumming we have 4 numbers: a, b, c, d, then
//   F(0) = 0a + 1b + 2c + 3d
//   F(1) = 3a + 0b + 1c + 2d
//   F(2) = 2a + 3b + 0c + 1d
//   F(3) = 1a + 2b + 3c + 0d
//
// We can see how F(n) transfrom to F(n+1) 
//   F(0) - F(1) = -3a + b + c + d  
//   F(1) - F(2) = a + -3b + c + d
//   F(2) - F(3) = a + b + -3c + d
//   F(3) - F(0) = a + b + c + -3d
//   
// So, we can tansfrom to the following experssion: 
// 
//   F(1) = F(0) - (a+b+c+d) + 4a  
//   F(2) = F[1] - (a+b+c+d) + 4b
//   F(3) = F[2] - (a+b+c+d) + 4c
// 
// Then, we can see this fomular:
//
//   F(n) = F(n-1) - sum(array) + len(array) * array[n-1]
//   
class Solution {
public:    
    int maxRotateFunction(vector<int>& A) {
        int sum = 0;
        int F = 0;
        for (int i=0; i < A.size(); i++) {
            sum += A[i];
            F += (i * A[i]);
        }
        int maxF = F;
        int len = A.size();
        //cout << "F(0) = " << maxF <<endl;
        for (int i=1; i< len; i++) {
            F = F - sum + len * A[i-1];
            //cout << "F(" << i << ") = " << F << endl;
            maxF = max(maxF, F);
        }
        
        return maxF;
    }
};