剑指 Offer II 012. 左右两边子数组的和相等https://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;
}
}
力扣https://leetcode.cn/problems/tvdfij/solution/zuo-you-liang-bian-zi-shu-zu-de-he-xiang-5j4r/时间复杂度O(n);空间复杂度O(1)。