138 子数组之和

原题网址:https://www.lintcode.com/problem/subarray-sum/description

描述

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

There is at least one subarray that it's sum equals to zero.

您在真实的面试中是否遇到过这个题?  

样例

给出 [-3, 1, 2, -3, 4],返回[0, 2] 或者 [1, 3].

标签
哈希表
子数组
 
 

思路1:设置两根指针start与end,每次查找固定start,end向右移动,遍历数组,找到和为0的return出去。这种属于暴力破解,时间复杂度O(n^2),空间复杂度O(1)。【时间596ms】

AC代码:

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> result;
      int n=nums.size();
      int sum;
      int start=0,end=0;
      for (;start<n;start++)
      {
          sum=nums[start];
          end=start;
          while(end<n-1&&sum!=0)//end<n-1防止end++后数组索引超出范围;
          {
              end++;
              sum+=nums[end];            
          }
          if (sum==0)
          {
              result.push_back(start);
              result.push_back(end);
              return result;
          }      
      }
    }
};

思路2:利用哈希表,创建一个map,记录从第一个元素开始到当前元素之和当前元素下标 的对应关系,若有一段子数组和为0,那么一定有一个和对应两个下标,此时就找到了一段连续的和为0的子数组,时间复杂度O(n),空间复杂度O(n)。转自此处

AC代码:

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> result;
    int n=nums.size();
    map<int,int> m;
    m[0]=-1;//初始化哈希表,以防和为零的序列正好从第一个元素开始;
    int sum=0;
    for (int i=0;i<n;i++)
    {
        sum+=nums[i];
        if (m.find(sum)!=m.end())
        {
            result.push_back(m[sum]+1);
            result.push_back(i);
            return result;
        }
        m.insert(pair<int,int>(sum,i));
    }
    }
};

如果不对map初始化m[0] = -1,则 if 程序块可以改成:

if (m.find(sum)!=m.end())
{
    result.push_back(m[sum]+1);
    result.push_back(i);
    return result;
}
else if (m.find(sum)==m.end()&&sum==0)
{
    result.push_back(0);
    result.push_back(i);
    return result;
}
else
{
    m.insert(pair<int,int>(sum,i));
}

 

参考:

https://blog.csdn.net/github_34248245/article/details/50903849

https://www.aliyun.com/jiaocheng/374347.html

https://blog.csdn.net/weiyanjie100/article/details/79026862

https://blog.csdn.net/ljlstart/article/details/48393675

 

 

原文地址:https://www.cnblogs.com/Tang-tangt/p/9145066.html