最长小于等于k的子数组

问题描述

题目链接
问题描述:

给定一个无序数组arr，其中元素可正、可负、可0。给定一个整数k，求arr所有的子数组中累加和小于或等于k的最长子数组长度
例如：arr = [3, -2, -4, 0, 6], k = -2. 相加和小于等于-2的最长子数组为{3, -2, -4, 0}，所以结果返回4
[要求]时间复杂度为𝑂(𝑛)，空间复杂度为𝑂(𝑛)
输入描述：
第一行两个整数N, k。N表示数组长度，k的定义已在题目描述中给出
第二行N个整数表示数组内的数
输出描述：输出一个整数表示答案
实例:

输入 : 5 -2
	  3  -2  -4  0  6
输出 : 4

解一 : 前缀和 + 二分查找

这个解比较容易想到, 构建前缀和, 然后遍历数组, 当遍历到i时, 前缀和为sum , 需要查找左边前缀和>=sum - k 的最左边的位置j, 当前i位置能得到的答案就是 i-j

解二 : 子数组最小累加和 + 滑动窗口

#include <bits/stdc++.h>
using namespace std;
#define ll long long 

void solve() {
    int n, k;
    cin >> n >> k;
    vector<int> a(n+1);
    for (int i = 1; i <= n; ++i) cin >> a[i];
    vector<int> MinSum(n + 1);//MinSum[i] 从i开始能得到的最小值
    vector<int> MinSumEnd(n + 1);//MinSumEnd[i] 从i开始能得到的最小值的子数组的结束位置
    MinSum[n] = a[n];
    MinSumEnd[n] = n;
    for (int i = n-1; i >= 1; --i) {
        if (MinSum[i + 1] < 0) {
            MinSum[i] = MinSum[i + 1] + a[i];
            MinSumEnd[i] = MinSumEnd[i + 1];
        }
        else {
            MinSum[i] = a[i];
            MinSumEnd[i] = i;
        }
    }
    int ans = 0;
    //i....end-1是sum已经确定<=k的区间, end位置是待扩充的位置
    //end不回退是因为i 开始已经能扩到 end-1 了, i+1位置最多能扩到的位置如果是 小于end-1的, 不会是答案(只关心最长的)
    for (int i = 1, end = 1, sum = 0; i <= n; ++i) {
        //end开始可以扩,那么就把end开始能得到的最小子数组扩进来, 得到新区间i....end.......MinSumEnd[end]
        while (end <= n && sum + MinSum[end] <= k) {
            sum += MinSum[end];
            end = MinSumEnd[end] + 1;
        }
        //end-1位置是从i开始能扩到的最远的距离,从end开始无法扩了, sum[i.....end-1] + MinSum[end] > k
        //end>i 说明从i开始是确确实实扩出去了, 收集答案, sum-=a[i]因为i位置马上出窗口了
        if (end > i) {
            ans = max(ans, end - i);
            sum -= a[i];
        }
        //否则end==i , 说明i位置开始的最小子数组>k, i位置无法扩
        else end = i + 1;
        
    }
    cout << ans << endl;
}

signed main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
   /*int t; cin >> t;
    while (t--)*/ solve();
}