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