最大子数组之和为k - LintCode

给一个数组nums和目标值k，找到数组中最长的子数组，使其中的元素和为k。如果没有，则返回0。

 注意事项
数组之和保证在32位有符号整型数的范围内

样例
给出数组 nums = [1, -1, 5, -2, 3]，k = 3，返回4

解析：
子数组[1, -1, 5, -2]的和为3，且长度最大

给出数组nums = [-2, -1, 2, 1]， k = 1，返回2

解析：
子数组[-1, 2]的和为1，且长度最大

挑战
能否用O(n)的时间复杂度完成？

思路
构建Map存放nums元素的值和位置。之后将nums[i]修改为nums[0]到nums[i]的和。如果nums[i]等于k，则子数组为nums[0,i]，长度为i+1。对于nums[i]，在Map中寻找nums[i]-k，若存在这样的元素，且其位置p小于i，新的子数组长度为i-p，取最小值。

#ifndef C911_H
#define C911_H
#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;
class Solution {
public:
    /**
    * @param nums: an array
    * @param k: a target value
    * @return: the maximum length of a subarray that sums to k
    */
    int maxSubArrayLen(vector<int> &nums, int k) {
        // Write your code here
        if (nums.empty())
            return 0;
        int res = 0;
        unordered_map<int, int> m;//存放nums中元素的值和位置
        int len = nums.size();
        //nums[i]表示从nums[0]到nums[i]的和
        for (int i = 1; i < len; ++i)
            nums[i] += nums[i - 1];
        for (int i = 0; i < len; ++i)
        {
            //重复元素,存放较小位置
            if (m.find(nums[i]) == m.end())
                m[nums[i]] = i;
        }
        //若nums[i]等于k,子数组为nums[0]到nums[i],长度为i + 1
        for (int i = len - 1; i >= 0; --i)
        {
            if (nums[i] == k)
            {
                res = i + 1;
                break;
            }
        }
        //在m中寻找nums[i]-k,如果存在,且m[nums[i]-k]的值小于等于i
        //说明在nums[0]到nums[i]中,存在子数组的和位k,子数组长度为i - m[nums[i] - k
        for (int i = 0; i < len; ++i)
        {
            if (m.find(nums[i] - k) != m.end() && m[nums[i] - k] <= i)
            {
                res = maxVal(res, i - m[nums[i] - k]);
            }
        }
        return res;
    }
    int maxVal(int a, int b)
    {
        return a > b ? a : b;
    }
};
#endif