未排序数组中累加和为给定值的最长子数组

1.题目描述

给定一个数组arr，该数组无序，但每个值均为正数，再给定一个正数k。求arr的所有子数组中所有元素相加和为k的最长子数组的长度

例如，arr = [1, 2, 1, 1, 1], k = 3

累加和为3的最长子数组为[1, 1, 1]，所以结果返回3

[要求]

时间复杂度为O(n)O(n)，空间复杂度为O(1)O(1)

输入描述

第一行两个整数N, k。N表示数组长度，k的定义已在题目描述中给出
第二行N个整数表示数组内的数

输出描述

输出一个整数表示答案

解题思路

给定一个正数数组，求其子数组，可以使用两个指针left,right指向开始位置0，则初始化子数组的和sum为 arr[0]，长度为right-left+1。

如果sum==k,则left和right确定的子数组符合要求，判断其长度，存储长度最大的值。

如果 sum<k，则right指针移动（因为数组中只有正数），如果sum>k,则指针left移动。移动时注意：right指针移动，是先right++，再sum+=arr[right];left指针移动，是先sum -=arr[left]，再left ++。

N,k = list(map(int,input().split()))
arr = list(map(int,input().split()))

len_ = 0
sum_ = arr[0]
left,right = 0,0
while right<N:
    if sum_ == k:
        len_ = max(len_,right-left+1)
        sum_ -= arr[left]
        left += 1
    elif sum_ < k:
        right += 1
        if right >= N:
            break
        sum_ += arr[right]
    else:
        sum_ -= arr[left]
        left += 1
print(len_)

2.题目描述

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

解题思路

当数组中同时存在正负数时，无法使用滑动的方法查找值为k的子数组，此时，应该换一种思路，用两个子数组相减的值来判断。

当我们遍历到i位置时，如果arr[0...i]-arr[0...j] == k，则说明arr[j...i] ==k。我们可以使用字典来记录arr[0...i]的值（重复出现的值记录最早出现的位置），就可以通过一次遍历得出结果。

代码实现

N,k = list(map(int,input().split()))
arr = list(map(int,input().split()))

len_ = 0
sub_dict = {}
sub_dict[0] = -1
sum_ = 0
for i in range(N):
    sum_ += arr[i]
    if sum_-k in sub_dict:
        len_ = max(len_,i-sub_dict[sum_-k])
    if sum_ in sub_dict:
        pass
    else:
        sub_dict[sum_] = i
print(len_)

3.题目描述

未排序数组中累加和为给定值的最长子数组系列问题补充1

给定一个无序数组arr，其中元素可正、可负、可0。求arr所有子数组中正数与负数个数相等的最长子数组的长度。

[要求]

时间复杂度为O(n)O(n)，空间复杂度为O(n)O(n)

解题思路：

该题目为上一题的变形，使用count表示前i位中正负数的个数，正数count++，负数count--,0不变，思路和上一题相同。

N = int(input())
arr = list(map(int,input().split()))

len_ = 0
count = 0
sub_dict = {}
sub_dict[0] = -1

for i in range(N):
    if arr[i] > 0:
        count += 1
    elif arr[i] <0:
        count -= 1
    else:
        pass
    if count in sub_dict:
        len_ = max(len_,i-sub_dict[count])
    if count not in sub_dict:
        sub_dict[count] = i
print(len_)