题目:
给定一个整数数组,找到和为零的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置
样例:
给出 [-3, 1, 2, -3, 4]
,返回[0, 2]
或者 [1, 3]
.
代码:
class Solution {
public:
/**
* @param nums: A list of integers
* @return: A list of integers includes the index of the first number
* and the index of the last number
*/
vector<int> subarraySum(vector<int> nums){
// write your code here
vector<int> ret;
int n;
if((n=nums.size())==0)
return ret;
map<int,int> mark;
mark[0]=-1;
int sum=0;
for(int i=0;i<n;++i){
sum+=nums[i];
if(mark.count(sum)){
ret.push_back(mark[sum]+1);
ret.push_back(i);
return ret;
}
mark[sum]=i;
}
return ret;
}
};
思想:这个题一开始不太会做,在网上看了别人的之后会做得。大体思路就是利用了一个map来记录从第一个元素到当前元素的和以及下标,如果这之后的元素和为0,sum不变,返回下标即可。