c_cpp 724.cpp

Posted 2021-05-14

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        if (nums.size() == 0) return -1;
        int total = 0;
        for(int i = 0; i < nums.size(); i++) {
            total += nums[i];
        }
        int l = 0, r = total;
        for(int j = 0; j < nums.size(); j++) {          
            //cout << "0> j: " << j << ", nums\[j-1\]: " << nums[j-1] << endl;
            //cout << "1> l: " << l << ", r: " << r << endl;
            if (j != 0) {
                l += nums[j-1];
            }
            r -= nums[j];
            //cout << "2> l: " << l << ", r: " << r << endl;
            if (l == r) {
                return j;
            }
        }
        return -1;
    }
};