leetcode-array-Find Pivot Index-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_40773388/article/details/104872441

本文深入解析LeetCode中一道关于数组中心索引的问题，介绍了一种有效的解决策略，通过两次遍历数组，先计算总和再逐项比较左右子数组之和，以确定是否存在中心索引。文章对比了两种算法思路，纠正了仅适用于正数数组的错误解法，并提供了一个通过所有测试案例的正确代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

做题回顾
https://leetcode.com/explore/learn/card/array-and-string/201/introduction-to-array/1144/

其实这道题不难，直接按他提示来就好。

就是遍历一遍数组求出数组的SUM，然后再遍历一遍数组，找到第一个(i左边数组的SUM)=SUM-(i左边数组的SUM)，返回这个i；找不到就返回-1.这里SUM不会溢出。

要注意有正负的情况

We can precompute prefix sums P[i] = nums[0] + nums[1] + … + nums[i-1]. Then for each index, the left sum is P[i], and the right sum is P[P.length - 1] - P[i] - nums[i].

我天真地只考虑了全是正数的情况，妄想只遍历一次得到 i ，就是简单地设置left = 0, right = size() - 1,从两头扫数组求和，计算两边值相等的时候就输出，结果（-1，-1，-1，0）这个用例就打趴了我的代码
原始错误代码如下.

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        int size = nums.size();
        if(size == 0 || size == 2){
            return -1;
        }
        if(size == 1){
            return 0;
        }
        int left = 0, right = size  -1;
        int sum_left = nums[left], sum_right = nums[right];
        for(int i = 0; i < (nums.size() - 3); i++){
            if(sum_left < sum_right){
                left++;
                sum_left += nums[left];
            }
            else{
                right--;
                sum_right += nums[right];
            }
        }
        if(sum_left != sum_right){
            return -1;
        }
        else{
            return (left + 1);
        }
    }
};

推倒重来

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

Accept~