Leetcode 209. 长度最小的子数组——go语言实现

文章目录

    • 一、题目描述
    • 二、代码实现
      • 方法一:暴力法
        • 解题思路
        • 代码实现
        • 复杂度分析
      • 方法二:滑动窗口 + 双指针
        • 解题思路
        • 代码实现
        • 复杂度分析
      • 方法三:前缀和 + 二分查找
        • 解题思路
        • 代码实现
        • 复杂度分析

一、题目描述

给定一个含有 n 个正整数的数组和一个正整数 target 。

找出该数组中满足其和 ≥ target 的长度最小的 连续子数组 [numsl, numsl+1, …, numsr-1, numsr] ,并返回其长度。如果不存在符合条件的子数组,返回 0 。

示例 1:

输入:target = 7, nums = [2,3,1,2,4,3]
输出:2
解释:子数组 [4,3] 是该条件下的长度最小的子数组。

示例 2:

输入:target = 4, nums = [1,4,4]
输出:1

示例 3:

输入:target = 11, nums = [1,1,1,1,1,1,1,1]
输出:0

进阶:如果你已经实现 O(n) 时间复杂度的解法, 请尝试设计一个 O(n log(n)) 时间复杂度的解法。

二、代码实现

方法一:暴力法

解题思路

  枚举数组 nums 中的每个下标作为子数组的开始下标,对于每个开始下标 i,需要找到大于或等于 i 的最小下标 j,使得从 nums[i] 到 nums[j] 的元素和大于或等于 s,并更新子数组的最小长度(此时子数组的长度是 j−i+1)。

代码实现
func minSubArrayLen1(target int, nums []int) int {
	minLen := math.MaxInt
	for i := 0; i < len(nums); i++ {
		sum := 0
		for j := i; j < len(nums); j++ {
			sum += nums[j]
			if sum >= target {
				minLen = min(minLen, j-i+1)
				break
			}
		}
	}
	if minLen == math.MaxInt {
		return 0
	}
	return minLen
}
func min(x, y int) int {
	if x < y {
		return x
	}
	return y
}

Leetcode 209. 长度最小的子数组——go语言实现_第1张图片

复杂度分析
  • 时间复杂度:O(n^2 ),其中 n 是数组的长度。需要遍历每个下标作为子数组的开始下标,对于每个开始下标,需要遍历其后面的下标得到长度最小的子数组。
  • 空间复杂度:O(1)。

方法二:滑动窗口 + 双指针

解题思路

  定义两个指针 left 和 right 分别表示子数组(滑动窗口)的开始位置和结束位置,维护变量 sum 存储子数组中的元素和(即从 nums[left] 到 nums[right] 的元素和)。

  初始状态下,left 和 right 都指向下标 0,sum 的值为 0。每一轮迭代,将 nums[right] 加到 sum,如果 sum≥target,则更新子数组的最小长度(此时子数组的长度是 right−left+1),然后将 nums[left] 从 sum 中减去并将 left 右移,直到 sum

代码实现
func minSubArrayLen(target int, nums []int) int {
	left, right := 0, 0
	minLen := math.MaxInt
	sum := 0
	for right < len(nums) {
		sum += nums[right]
		for sum >= target {
			minLen = min(minLen, right-left+1)
			sum -= nums[left]
			left++
		}
		right++
	}
	if minLen == math.MaxInt {
		return 0
	}
	return minLen
}
func min(x, y int) int {
	if x < y {
		return x
	}
	return y
}

Leetcode 209. 长度最小的子数组——go语言实现_第2张图片

复杂度分析
  • 时间复杂度:O(n),其中 n 是数组的长度。指针 left 和 right 最多各移动 n 次。

  • 空间复杂度:O(1)。

方法三:前缀和 + 二分查找

解题思路

  方法一采用暴力枚举,时间复杂度是 O(n^2 ),因为在确定每个子数组的开始下标后,找到长度最小的子数组需要 O(n) 的时间。如果使用二分查找,则可以将时间优化到 O(logn)。

  每个下标开始的这个遍历过程计算了很多重复的区间,比如1,2,3,4 。以1为下标时计算了+2、+3、+4 ; 以2为下标时计算了+3、+4,像这种避免区间和重复计算的优化方法,我们想到了前缀和,可以O(1)时间迅速得到任意区间的和。

  然后 ,我们可以很容易得改良问题为,求s[j] - s[i] >=target ,可是这种做法如果不加改变,就是在前缀和数组上进行类似方法一的暴力枚举,枚举每一个 i 后面的下标 j 。我们发现稍作变化,像这种线性的求值问题,联合二分查找可以做到 求 s[j] >=s[i]+target,将时间优化到 O(logn)。

代码实现
func minSubArrayLen(target int, nums []int) int {
	ans := math.MaxInt32
	sums := make([]int, 1)
	for i := 1; i <= len(nums); i++ {
		sums = append(sums, sums[i-1]+nums[i-1])
	}
	for i := 1; i <= len(nums); i++ {
		num := target + sums[i-1]
		bound := sort.SearchInts(sums, num)
		if bound <= len(nums) {
			ans = min(ans, bound-(i-1))
		}
	}
	if ans == math.MaxInt32 {
		return 0
	}
	return ans
}
func min(x, y int) int {
	if x < y {
		return x
	}
	return y
}

Leetcode 209. 长度最小的子数组——go语言实现_第3张图片

复杂度分析
  • 时间复杂度:O(nlogn),其中 n 是数组的长度。需要遍历每个下标作为子数组的开始下标,遍历的时间复杂度是 O(n),对于每个开始下标,需要通过二分查找得到长度最小的子数组,二分查找得时间复杂度是 O(logn),因此总时间复杂度是 O(nlogn)。

  • 空间复杂度:O(n),其中 n 是数组的长度。额外创建数组 sums 存储前缀和。

你可能感兴趣的:(面试高频算法题,Leetcode,leetcode,golang,算法)