您的位置:首页 > 其它

lintcode-子数组之和-138

2015-09-12 14:02 337 查看
给定一个整数数组,找到和为零的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置



样例


给出[-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;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: