724. Find Pivot Index

问题：

求给定数组的对称轴index。（其左边元素之和=右边元素之和）

若不存在则返回-1，若存在多个，返回最左边的对称轴。

Example 1:
Input: 
nums = [1, 7, 3, 6, 5, 6]
Output: 3
Explanation: 
The sum of the numbers to the left of index 3 (nums[3] = 6) is equal to the sum of numbers to the right of index 3.
Also, 3 is the first index where this occurs.
 

Example 2:
Input: 
nums = [1, 2, 3]
Output: -1
Explanation: 
There is no index that satisfies the conditions in the problem statement.
 

Note:
The length of nums will be in the range [0, 10000].
Each element nums[i] will be an integer in the range [-1000, 1000].

　　

解法：

首先求出所有元素之和 rightsum

再从0开始遍历数组，求 leftsum

rightsum-=nums[i]

如果leftsum==rightsum，则返回 i，

循环结束，返回-1。

 

代码参考：

auto fastio = []() {  //加速方法：去掉io绑定和不需要的功能。这里用到了c++11的Lambda

    ios::sync_with_stdio(0);
    cin.tie(NULL), cout.tie(NULL);
    return 0;
}();

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int leftsum=0, rightsum=0;
        for(int i:nums) rightsum+=i;
        for(int i=0; i<nums.size(); i++){
            rightsum-=nums[i];
            if(leftsum==rightsum) return i;
            leftsum+=nums[i];
        }
        return -1;
    }
};

 

另一种方法，

leftsum*2==sum-nums[i]

以此判断中轴。

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int leftsum=0, sum=0;
        for(int i:nums) sum+=i;
        for(int i=0; i<nums.size(); i++){
            if(leftsum*2==sum-nums[i]) return i;
            leftsum+=nums[i];
        }
        return -1;
    }
};

 

参考：

加速方法解释：https://blog.csdn.net/qq_32320399/article/details/81518476