题目:
给定一个整数类型的数组 nums,请编写一个能够返回数组“中心索引”的方法。
我们是这样定义数组中心索引的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。
如果数组不存在中心索引,那么我们应该返回 -1。如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
暴力解法:
class Solution {
public int pivotIndex(int[] nums) {
for(int i=0;i<nums.length;i++){
int sum1=0;
int sum2=0;
//计算左侧所有元素的和
for(int j=i-1;j>=0;j--){
sum1+=nums[j];
}
//计算右侧所有元素的和
for(int j=i+1;j<nums.length;j++){
sum2+=nums[j];
}
//判断左侧和与右侧和是否相等,相等则返回i值,循环结束
if(sum1==sum2){
return i;
}
}
return -1; //若无符合条件的中心索引,则在数组遍历结束后返回-1
}
}
优化:
1.sumLeft + sumRight + nums[p] = sumTotal;
2.sumLeft = sumRight;
3.可以得出 sumLeft * 2 + nums[p] = sumTotal;
class Solution {
public int pivotIndex(int[] nums) {
int sumTotal = 0;
int sumLeft = 0;
for (int x: nums){
sumTotal += x;
}
for (int i = 0; i < nums.length; i++) {
if (sumLeft*2 == sumTotal - nums[i]){
return i;
}
sumLeft += nums[i];
}
return -1;
}
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-pivot-index