牛客网. 未排序正数数组中累加和为给定值的最长子数组的长度

georgeandgeorge

于 2021-06-02 09:44:13 发布

阅读量155

点赞数 1

分类专栏： oj系列文章标签： leetcode 数据结构算法 c++

本文链接：https://blog.csdn.net/georgeandgeorge/article/details/117460194

版权

oj系列专栏收录该内容

84 篇文章 0 订阅

订阅专栏

题目概述

解题思路

我的思路是用双指针法，记录当前子数组的长度和总和，然后动态地调整即可。

方法性能

#include<iostream>
using namespace std;

int main()
{
    int N, K;
    cin >> N >> K;
    int *A = new int[N];

    for (int i = 0; i < N;++i)
        cin >> A[i];

    int sta_i = 0, end_i = 0, sum = A[0], max_len = 0;
    while(end_i < N)
    {
        if(sum == K)
        {
            max_len = max(max_len, end_i - sta_i + 1);
            if(sta_i == end_i)
            {
                end_i++;
                sum += A[end_i];
            }
            sum -= A[sta_i];
            sta_i++;
        }
        else if(sum < K)
        {
            end_i++;
            sum += A[end_i];
        }
        else
        {
            sum -= A[sta_i];
            sta_i++;
            if(sta_i > end_i)
            {
                end_i++;
                sum += A[end_i];
            }
        }
    }

    cout << max_len << endl;

    return 0;
}