给定一个整数数组,找到和为零的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置。 假定一定存在这样的字数组。 样例 给出 [-3, 1, 2, -3, 4],返回[0, 2] 或者 [1, 3].
遍历数组,求第一个元素到达当前元素的累加和,如果累加和出现0,或者累加和有重复出现,那么就找到和为0的子数组了。
num | -3 | 2 | 1 | 2 | 3 |
---|---|---|---|---|---|
sum | -3 | -1 | 0 | 2 | 5 |
index | 0 | 1 | 2 | 3 | 4 |
这里累加到1的时候出现了0,那么说明从第一个数到1这个数的累加和是0,很好理解。
num | 1 | -2 | -3 | 3 | 2 |
---|---|---|---|---|---|
sum | 1 | -1 | -4 | -1 | 1 |
index | 0 | 1 | 2 | 3 | 4 |
这里index=0和index=4的两个累加和相同了,说明这两个累计和之间的所有数的和是0,其实也很好理解,只是没想到可以这么做。 代码写起来就比较简单了,为了代码统一,先把0存起来,因为一下子要存两个数,所以这里用了map。还可以很简单的利用map的find函数。
vector<int> subarraySum(vector<int> nums){
// write your code here
int sum = 0;
vector<int> ret;
unordered_map<int, int> map;
map[0] = -1; //这个是把0先加进去了,
for(int i=0; i<nums.size();i++)
{
sum += nums[i];
if(map.find(sum) != map.end())
{ cout<<map.find(sum)->first<<endl;
ret.push_back(map[sum] +1);
ret.push_back(i);
return ret;
}
map[sum] = i;
}
return ret;
}