题目描述
给定一个无序数组arr, 其中元素可正、可负、可0。给定一个整数k,求arr所有子数组中累加和为k的最长子数组长度
输入描述:
第一行两个整数N, k。N表示数组长度,k的定义已在题目描述中给出
第二行N个整数表示数组内的数
输出描述:
输出一个整数表示答案
示例1
输入
5 0
1 -2 1 1 1
输出
3
思路:
对于数组arr[i],建立和数组arrS[i]表示从序号0到i的值的和。则对于数组arr[i],序号i+1到j的子数组和可以表示为arrS[j]-arrS[i],题目转化为求arrS[j]-k=arrS[i]是否存在。
利用unordered_map中的hash表可以快速查找,复杂度为O(1)。
#include<iostream>
#include<unordered_map>
using namespace std;
int main()
{
int N, k;
int sum = 0;
int max = 0;
cin >> N >> k;
unordered_map<int, int> arrS;
arrS[0] = -1;
for (int i = 0; i<N; ++i)
{
int temp;
cin >> temp;
sum += temp;
if (arrS.find(sum - k) != arrS.end())
max = max>(i - arrS[sum - k]) ? max : (i - arrS[sum - k]);
if (arrS.find(sum) == arrS.end())
arrS[sum] = i;
}
cout << max << endl;
return 0;
}
京公网安备 11010502036488号