牛客题霸 [ 未排序数组中累加和为给定值的最长子数组长度] C++题解/答案

牛客题霸 [ 未排序数组中累加和为给定值的最长子数组长度] C++题解/答案

题目描述

给定一个无序数组arr, 其中元素可正、可负、可0。给定一个整数k，求arr所有子数组中累加和为k的最长子数组长度

题解：

先求出前缀和
然后用map来记录第i位的前缀和
要找和为k的子数组
我们直接看(第i位前缀和+k)是否存在，如果存在map的值指向的第j位，那长度就是j-i+1
取最大值
当然还有可能前缀和本身就是k，记得不要忘了这种情况

代码：

#include<unordered_map>
class Solution {
public:
    /**
     * max length of the subarray sum = k
     * @param arr int整型vector the array
     * @param k int整型 target
     * @return int整型
     */
    int maxlenEqualK(vector<int>& arr, int k) {
        // write code here
        if(arr.size()==0)return 0;
       vector<int>w(arr.size(),0);
        w[0]=arr[0];
        for(int i=1;i<arr.size();i++)
        {
            w[i]=w[i-1]+arr[i];
        }
        int maxx=0;
        unordered_map<int,int> ump;
        for(int i=0;i<arr.size();i++){
            ump[w[i]]=i;
        }
         
        int ans=0;
        for(int i=0;i<arr.size();i++){
            if(ump.count(w[i]+k)){
                ans =max(ans, ump[w[i]+k]-i);
            }
        }
        if(ump.count(k)) ans = max(ans, ump[k]+1);
        return ans;
    }
};