lintcode-子数组之和-138

给定一个整数数组,找到和为零的子数组。你的代码应该返回满足要求的子数组的起始位置和结束位置


样例

给出[-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;
    }
};


你可能感兴趣的:(lintcode-子数组之和-138)