力扣刷题练习六【724.寻找数组的中心下标】

最新推荐文章于 2024-08-18 22:40:28 发布

danaaaa

最新推荐文章于 2024-08-18 22:40:28 发布

阅读量381

点赞数 3

分类专栏：算法力扣额外题目解答文章标签： leetcode 算法职场和发展 c++

本文链接：https://blog.csdn.net/danaaaa/article/details/140101237

版权

算法力扣额外题目解答专栏收录该内容

9 篇文章 0 订阅

订阅专栏

前言

数组篇练习题。
练习六【724.寻找数组的中心下标】

一、题目阅读

给你一个整数数组 nums ，请计算数组的中心下标。

数组中心下标是数组的一个下标，其左侧所有元素相加的和等于右侧所有元素相加的和。

如果中心下标位于数组最左端，那么左侧数之和视为 0 ，因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。

如果数组有多个中心下标，应该返回最靠近左边的那一个。如果数组不存在中心下标，返回 -1 。

示例 1：

输入：nums = [1, 7, 3, 6, 5, 6]
输出：3
解释：
中心下标是 3 。
左侧数之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 ，
右侧数之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ，二者相等。

示例 2：

输入：nums = [1, 2, 3]
输出：-1
解释：
数组中不存在满足此条件的中心下标。

示例 3：

输入：nums = [2, 1, -1]
输出：0
解释：
中心下标是 0 。
左侧数之和 sum = 0 ，（下标 0 左侧不存在元素），
右侧数之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。

提示：

1 <= nums.length <= 104
-1000 <= nums[i] <= 1000

二、尝试实现

思路一

暴力解法。

代码实现一

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        if(nums.size() == 1){
            return 0;
        }

        for(int i = 0 ;i < nums.size();i++){
            int sumleft = 0;
            int sumright = 0;
            for(int left = 0;left < i;left++){
                sumleft += nums[left];
            }
            for(int right = i+1;right < nums.size();right++){
                sumright += nums[right];
            }
            if(sumleft == sumright){
                return i;
            }
        }

        return -1;
};

思路二

考虑用双指针。使用双指针需要考虑指针从哪开始？

（弯路）如果把指针放在两端，往中间靠拢，等待相遇？或者判断求和一样？其实不符合题意，两边之和，处理不对，行不通。
用两个滑动窗口：

代码实现二

class Solution {
public:
    int pivotIndex(vector<int>& nums) {
        if(nums.size() == 1){
            return 0;
        }
        int sumright = 0;
        for(int i = nums.size()-1;i > 0;i--){
            sumright += nums[i];
        }
        int sumleft = 0;
        //两个滑动窗口
        for(int i = 0;i < nums.size();i++){   
            if(sumleft == sumright){
                return i;
            }else if(i != nums.size()-1){
                sumleft += nums[i];
                sumright -= nums[i+1];
            }
        }
        return -1;
    }   
};

注意点：

for循环只需要判断相不相等，没有else if(sumleft < sumright)这种判断，因为nums[i]有正有负，无法知道是左边小还是右边小，只能不相等的时候移动即可。
为了使nums[i+1]合法，不能在for条件判断处：i < nums.size()-1，这样漏掉i == nums.size()-1。只能在else if出加个条件，顺利处理。