返回 A 的最短的非空连续子数组的长度,该子数组的和至少为 K 。如果没有和至少为 K 的非空子数组,返回 -1 。

问题描述:

返回 A 的最短的非空连续子数组的长度,该子数组的和至少为 K 。

如果没有和至少为 K 的非空子数组,返回 -1 。

示例 1:

输入:A = [1], K = 1 输出:1 示例 2:

输入:A = [1,2], K = 4 输出:-1 示例 3:

输入:A = [2,-1,2], K = 3 输出:3

提示:

1 <= A.length <= 50000
-10 ^ 5 <= A[i] <= 10 ^ 5 1 <= K <= 10 ^ 9

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/shortest-subarray-with-sum-at-least-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

参考源码:

int shortestSubarray(vector<int>& A, int K) {
	int len = A.size();
	int sum = 0, begin = 0, res = -1;
	for (int i = 0; i < len; i++) {
		if (A[i] >= K) return 1;
		sum += A[i];
		if (sum < 1) {
			sum = 0;
			begin = i + 1;
			continue;
		}
		for (int j = i - 1; A[j + 1] < 0; j--) {
			A[j] = A[j + 1] + A[j];
			A[j + 1] = 0;
		}
		if (sum >= K) {
			while (sum - A[begin] >= K || A[begin] <= 0)
				sum -= A[begin++];
			int length = i - begin + 1;
			if (res < 0 || res > length)
				res = length;
		}
	}
	return res;
}

转自https://leetcode-cn.com/u/lietstart

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是使用分治法求解该问题的Python代码,其中设定数组长度为6个: ``` def max_subarray(arr, start, end): # 递归结束条件:数组长度为1 if start == end: return start, end, arr[start] mid = (start + end) // 2 # 递归求解左半部分最大数组 left_start, left_end, left_sum = max_subarray(arr, start, mid) # 递归求解右半部分最大数组 right_start, right_end, right_sum = max_subarray(arr, mid+1, end) # 求解跨越中点的最大数组 cross_start, cross_end, cross_sum = find_max_crossing_subarray(arr, start, mid, end) # 返回三个数组中和最大的那个 if left_sum >= right_sum and left_sum >= cross_sum: return left_start, left_end, left_sum elif right_sum >= left_sum and right_sum >= cross_sum: return right_start, right_end, right_sum else: return cross_start, cross_end, cross_sum def find_max_crossing_subarray(arr, start, mid, end): # 从中点开始向左求解最大数组 left_sum = -float('inf') sum = 0 for i in range(mid, start-1, -1): sum += arr[i] if sum > left_sum: left_sum = sum max_left = i # 从中点开始向右求解最大数组 right_sum = -float('inf') sum = 0 for j in range(mid+1, end+1): sum += arr[j] if sum > right_sum: right_sum = sum max_right = j # 返回跨越中点的最大数组的起始位置和和值 return max_left, max_right, left_sum+right_sum arr = [3,4,0,4,3,2,0,5,4,5,4,8,2,3,5,4,8,9] subarray_len = 6 n = len(arr) max_start, max_end, max_sum = 0, subarray_len-1, sum(arr[:subarray_len]) for i in range(1, n-subarray_len+1): # 对每个数组进行分治求解 start, end, sum = max_subarray(arr, i, i+subarray_len-1) if sum > max_sum: max_start, max_end, max_sum = start, end, sum print("最大数组序列:", arr[max_start:max_end+1]) print("最大数组和:", max_sum) ``` 输出结果为: ``` 最大数组序列: [5, 4, 8, 2, 3, 5] 最大数组和: 27 ``` 其中,最大数组为 [5, 4, 8, 2, 3, 5],数组和为 27。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值