版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/majichen95/article/details/83035312
给定一个整数数组nums
,编写一个返回此数组的“中心索引”的方法。
我们将中心索引定义为:中心索引左边的数字之和等于中心索引右边的数字之和。
如果不存在这样的中心索引,我们应该返回-1。 如果有多个中心索引,则应返回最左侧的那个。
样例
样例1:
输入:
nums = [1, 7, 3, 6, 5, 6]
输出: 3
解释:
索引3 (nums[3] = 6)左侧所有数之和等于右侧之和。
并且3是满足条件的第一个索引。
样例2:
输入:
nums = [1, 2, 3]
输出: -1
解释:
并没有满足条件的中心索引。
注意事项
nums
的长度范围为[0, 10000]
.nums[i]
中每个元素都是[-1000, 1000]
范围内的整数.
解题思路:
通常的想法是暴力求解,但这样时间复杂度是O(n^2),会超时。
可以观察到每一次移动中心索引,变化的都只是此索引对应的值,无非每移动一次,左边的和加一次,右边的和减一次,所以可以只遍历一次,通过遍历索引的变化来变化两边的和,时间复杂度为O(n)。
public class Solution {
/**
* @param nums: an array
* @return: the "pivot" index of this array
*/
public int pivotIndex(int[] nums) {
// Write your code here
int sumLeft = 0;//初始左边的和为0
int sumRight = 0;//初始右边的和为sum(nums)
for(int num : nums)
sumRight += num;
for(int i=0 ; i<nums.length ; i++){
sumRight -= nums[i];
if(sumRight == sumLeft)
return i;
sumLeft += nums[i];
}
return -1;
}
}