Tony's Log

Algorithms, Distributed System, Machine Learning

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

Nothing special. Point is, how clean can you do it?

class Solution 
{
public:
    int maxSubArrayLen(vector<int>& nums, int k) 
    {
        int n = nums.size();
        
        //    Partial Sum
        vector<int> asum(n);
        partial_sum(nums.begin(), nums.end(), asum.begin());

        int ret = 0;
        unordered_map<int, int> hm; // asum - min inx
        hm[0] = -1;
        for (int i = 0; i < n; i++)
        {
            int tgt = asum[i] - k;
            if (hm.count(tgt))    ret = max(ret, i - hm[tgt]);
            if (!hm.count(asum[i]))    hm[asum[i]] = i;
        }
        return ret;
    }
};
posted on 2016-01-06 18:33  Tonix  阅读(160)  评论(0编辑  收藏  举报