前言
数组篇练习题。
练习 六【724.寻找数组的中心下标】
一、题目阅读
给你一个整数数组 nums ,请计算数组的 中心下标 。
数组 中心下标 是数组的一个下标,其左侧所有元素相加的和等于右侧所有元素相加的和。
如果中心下标位于数组最左端,那么左侧数之和视为 0 ,因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。
如果数组有多个中心下标,应该返回 最靠近左边 的那一个。如果数组不存在中心下标,返回 -1 。
示例 1:
输入:nums = [1, 7, 3, 6, 5, 6]
输出:3
解释:
中心下标是 3 。
左侧数之和 sum = nums[0] + nums[1] + nums[2] = 1 + 7 + 3 = 11 ,
右侧数之和 sum = nums[4] + nums[5] = 5 + 6 = 11 ,二者相等。
示例 2:
输入:nums = [1, 2, 3]
输出:-1
解释:
数组中不存在满足此条件的中心下标。
示例 3:
输入:nums = [2, 1, -1]
输出:0
解释:
中心下标是 0 。
左侧数之和 sum = 0 ,(下标 0 左侧不存在元素),
右侧数之和 sum = nums[1] + nums[2] = 1 + -1 = 0 。
提示:
1 <= nums.length <= 104
-1000 <= nums[i] <= 1000
二、尝试实现
思路一
暴力解法。
代码实现一
class Solution {
public:
int pivotIndex(vector<int>& nums) {
if(nums.size() == 1){
return 0;
}
for(int i = 0 ;i < nums.size();i++){
int sumleft = 0;
int sumright = 0;
for(int left = 0;left < i;left++){
sumleft += nums[left];
}
for(int right = i+1;right < nums.size();right++){
sumright += nums[right];
}
if(sumleft == sumright){
return i;
}
}
return -1;
};
思路二
考虑用双指针。使用双指针需要考虑指针从哪开始?
- (弯路)如果把指针放在两端,往中间靠拢,等待相遇?或者判断求和一样?其实不符合题意,两边之和,处理不对,行不通。
- 用两个滑动窗口:
代码实现二
class Solution {
public:
int pivotIndex(vector<int>& nums) {
if(nums.size() == 1){
return 0;
}
int sumright = 0;
for(int i = nums.size()-1;i > 0;i--){
sumright += nums[i];
}
int sumleft = 0;
//两个滑动窗口
for(int i = 0;i < nums.size();i++){
if(sumleft == sumright){
return i;
}else if(i != nums.size()-1){
sumleft += nums[i];
sumright -= nums[i+1];
}
}
return -1;
}
};
注意点:
- for循环只需要判断相不相等,没有else if(sumleft < sumright)这种判断,因为nums[i]有正有负,无法知道是左边小还是右边小,只能不相等的时候移动即可。
- 为了使nums[i+1]合法,不能在for条件判断处:i < nums.size()-1,这样漏掉i == nums.size()-1。只能在else if出加个条件,顺利处理。
三、参考思路
和思路二窗口推进同理,不过处理上有区别:
用总和sum、leftsum确定rightsum。
四、总结
双指针(滑动窗口)练习+1.
(欢迎指正,转载标明出处)