给定一个整数数组,找到和为零的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置
样例
给出[-3, 1, 2, -3, 4],返回[0, 2] 或者 [1, 3].
第一种思路:利用两个for循环,每次取出一个元素依次与后面的元素相加,时间复杂度是O(n^2)。
第二种思路:利用一个map记录从第一个元素开始到当前元素之和以及当前元素的下标,如果后面的连续元素之和为零那么sum不变,此时就找到了和为零的连续序列,时间复杂度是O(n)
class Solution {
public:
vector<int> subarraySum(vector<int> nums){
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;
}
};