牛客题霸 [ 未排序数组中累加和为给定值的最长子数组长度] 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 hereif(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;}
};