leecode错题记录(1)

剑指 Offer II 012. 左右两边子数组的和相等icon-default.png?t=M7J4https://leetcode.cn/problems/tvdfij/

 这是一道难度简单的题。看到它的第一眼首先立刻想到的是双指针。分别设置两个指针指向数组的一头一尾,然后向中间收缩,收缩过程中分别计算两端一遍历元素的和,知道两指针相遇,若两边和相等则返回下标,否则返回-1。

class Solution {
    public int pivotIndex(int[] nums) {
        int leftCount=0, rightCounut=0;
        int left=0, right=nums.length-1;
        while(left<right ){
            leftCount += nums[left];
            while(leftCount>rightCounut){
                rightCounut += nums[right];
                right--;
            }
            left++;
        }
        if(leftCount==rightCounut){
            return right;
        }
        return -1;
    }
}

代码写出来用了不到五分钟,而且测试用例也过了,然而后面发现并不适用与返回结果为0的这种情况,例如下面这个用例,但是我又不知道怎么修改,有想法的朋友可以私信。

翻看了一下官网解法,才发现自己愚钝了,这里可以直接用求和的方法来做,而且Java提供了一个工具类可以直接对数组进行求和(现在才知道), Arrays.stream(nums).sum()可以直接对数组进行求和,求和后直接从数组头部进行遍历,若两倍的左边元素的和加指针指向元素的数值等于整个数组的和则返回指针指向元素的下标,思想也很简单,完全没有必要用到双指针,增大系统开销。

以为为官方解法:

class Solution {
    public int pivotIndex(int[] nums) {
        int total = Arrays.stream(nums).sum();
        int sum = 0;
        for (int i = 0; i < nums.length; ++i) {
            if (2 * sum + nums[i] == total) {
                return i;
            }
            sum += nums[i];
        }
        return -1;
    }
}

力扣icon-default.png?t=M7J4https://leetcode.cn/problems/tvdfij/solution/zuo-you-liang-bian-zi-shu-zu-de-he-xiang-5j4r/时间复杂度O(n);空间复杂度O(1)。

  • 4
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值