程序员代码面试指南第二版 124.未排序正数数组中累加和为给定值的最长子数组的长度

welcome to my blog

程序员代码面试指南第二版 124.未排序正数数组中累加和为给定值的最长子数组的长度

题目描述

题目描述
给定一个数组arr，该数组无序，但每个值均为正数，再给定一个正数k。求arr的所有子数组中所有元素相加和为k的最长子数组的长度
例如，arr = [1, 2, 1, 1, 1], k = 3
累加和为3的最长子数组为[1, 1, 1]，所以结果返回3
[要求]
时间复杂度为O(n)，空间复杂度为O(1)

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

输出描述:
输出一个整数表示答案

示例1

输入
5 3
1 2 1 1 1

输出
3

第一次做; while循环的写法; 核心:sum的初始化, sum的含义, sum表示子数组arr[left,right]的累加和; 看起来是for循环版本的简洁些

import java.util.Scanner;


public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String[] str = sc.nextLine().split(" ");
        int n = Integer.parseInt(str[0]);
        int k = Integer.parseInt(str[1]);
        str = sc.nextLine().split(" ");
        int[] arr = new int[n];
        for(int i=0; i<n; i++){
            arr[i] = Integer.parseInt(str[i]);
        }
        //sum表示子数组arr[left,right]的累加和
        int left=0, right=0, len=0, sum=arr[0];
        /*
        因为数组元素全是正数,所以可以使用双指针
        先移动右指针直到sum>k, 判断是否更新len, 然后移动左指针,直到sum<k
        重复上述指针移动过程
        */
        //注意sum初始化为arr[0]
        while(right<n){
            if(sum==k){
                len = Math.max(len, right-left+1);
                sum = sum - arr[left];
                //以left+1这个位置作为新的子数组的开头
                left++;
            }
            else if(sum<k){
                right++;
                if(right==n)
                    break;
                sum = sum + arr[right];
                
            }
            else{
                //以left+1这个位置作为新的子数组的开头
                sum = sum - arr[left];
                left++;
            }
        }
        System.out.print(len);
    }
}

第一次做; 双指针; 初始时left=0,right=0;先移动右指针再移动左指针; 核心:考查的是以left开头的子数组; 细节:sum表示子数组arr[left,right]的和

import java.util.Scanner;


public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String[] str = sc.nextLine().split(" ");
        int n = Integer.parseInt(str[0]);
        int k = Integer.parseInt(str[1]);
        str = sc.nextLine().split(" ");
        int[] arr = new int[n];
        for(int i=0; i<n; i++){
            arr[i] = Integer.parseInt(str[i]);
        }
        //
        int left=0, right=0, len=0, sum=0;
        /*
        因为数组元素全是正数,所以可以使用双指针
        先移动右指针直到sum>k, 判断是否更新len, 然后移动左指针,直到sum<k
        重复上述指针移动过程
        */
        for(int j=0; j<n; j++){
            sum = sum + arr[j];
            if(sum<k)
                continue;
            else if(sum==k)
                len = Math.max(len, j-left+1);
            else{
                while(sum>k){
                    sum = sum - arr[left];
                    left++;
                }
                if(sum==k)
                    len = Math.max(len, j-left+1);
            }
        }
        System.out.print(len);
    }
}