每日一题
724. 寻找数组的中心索引](https://leetcode-cn.com/problems/find-pivot-index/)
一 题目描述
给定一个整数类型的数组 nums,请编写一个能够返回数组 “中心索引” 的方法。
我们是这样定义数组 中心索引 的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。
如果数组不存在中心索引,那么我们应该返回 -1。如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-pivot-index
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二 思路
- 一定是从第二个开始,到倒数第二个结束
- 停止的条件:左边的和 == 右边的和 || 中间坐标的位置到了最后一个
三 可能需要的API
求和:
- 显然这个题目是不可能躲过求和的,当然直接遍历求和会很快,但是既然我是用java的话,应该考虑以下求和函数
四 实现思路
- 利用arraylist进行求和
- 求和之后还要减去前两个值作为第一次比较的右边和,将第一个值作为左边的和,将第二个值作为暂时的“中心索引”并进行判断
- 中心索引的位置一直右移,移动之后左边的和并入新的节点,右边的值减去新的中心索引,直到左边的和等于右边的和或者索引的位置到达最后一个返回-1
五 代码
class Solution {
public int pivotIndex(int[] nums) {
//增强鲁棒性
if(nums.length == 0){
return -1;
}
//
if(nums.length == 1){
return 0;
}
//求右边和
int sumRight = 0;
for(int i = 1; i < nums.length; i++){
sumRight += nums[i];
}
//左边的和
int sumLeft = 0;
//中间索引的位置
int centerIndex = 0;
//开始比较
while(centerIndex <= nums.length - 1){
if(sumLeft == sumRight){
return centerIndex;
}
sumLeft += nums[centerIndex];
if(centerIndex + 1 <= nums.length - 1) {
sumRight -= nums[centerIndex + 1];
}
centerIndex ++;
}
//找不到的话返回-1
return -1;
}
}
六 反思
- 这里和一开始的时候理解是有出入的,因为“中心索引”的两侧可以是没有值的,比如没有左边、没有右边或者两边都没有
- 程序的编写一定要考虑鲁棒性,无论何时都应该考虑极端的输入情况,比如空输入
- 注意临界点的判断,比如在此题目中centerIndex到达最右侧的时候,如果再按照原来的处理思路进行的话,就会报outOfIndex错误
- 没用到求和API,没必要,但是需要注意的一点是int数组转list的时候的一些注意事项