题目描述
给定一个无序数组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; }