题目描述
输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。
序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。
示例 1:
输入:target = 9
输出:[[2,3,4],[4,5]]
示例 2:输入:target = 15
输出:[[1,2,3,4,5],[4,5,6],[7,8]]
限制:1 <= target <= 10^5
我的解题思路:双指针+滑动窗口,利用双指针确定连续正数的两个边界,异动两个边界保证所以满足条件的序列都遍历到,详细题解见代码备注
public int[][] findContinuousSequence(int target) {
// 边界值处理,目标值小于3时无可匹配数值直接返回
if (target < 3) {
return new int[0][];
}
// 定义双指针--指向目标值取值两个边界
int left = 1, right = 2;
// 记录两边界内数值的和
int count = 3;
// 定义list便于存储不确定数量的一维数组
List<int[]> list = new ArrayList<>();
// 循环处理
while (left <= right && left <= target - 1) {
// 当至少有两个数字且范围内数值之和等于目标值时
if (right > left && count == target) {
// 新生成一维数组记录该值
int[] array = findCount(left, right);
list.add(array);
// 如右边界没超过目标值--继续寻找其他可能性
if (right < target) {
right ++;
count += right;
}
} else if (count > target) {
// 当边界值内数值之和大于目标值则数值之和减去原有左边界值,且左边界右异动一格
count -= left;
left++;
} else if (count < target && right < target) {
// 当边界值内数值之和小于目标值则右边界右异动一格,且数值之和加上新加入的右边界值
right++;
count += right;
}
}
// 将List转为二维数组返回
int length = list.size();
int[][] arr = new int[length][];
for(int i = 0; i < length; i++) {
arr[i] = list.get(i);
}
return arr;
}
public int[] findCount(int left, int right) {
int length = right - left + 1;
int[] array = new int[length];
for (int i = 0; i < length; i++) {
array[i] = left;
left++;
}
return array;
}
大神简洁版代码:思路一致,但是代码更加简洁值得学习
算法流程:
- 初始化: 左边界 i = 1 ,右边界 j = 2,元素和 s = 3 ,结果列表 res ;
- 循环: 当i≥j 时跳出;
- 当 s > target时: 向右移动左边界 i = i + 1,并更新元素和 s ;
- 当 s < target 时: 向右移动右边界 j = j + 1,并更新元素和 s ;
- 当 s = target 时: 记录连续整数序列,并向右移动左边界 i = i + 1;
- 返回值: 返回结果列表 res ;
public int[][] findContinuousSequence2(int target) {
int i = 1, j = 2, s = 3;
List<int[]> res = new ArrayList<>();
while(i < j) {
if(s == target) {
int[] ans = new int[j - i + 1];
for(int k = i; k <= j; k++)
ans[k - i] = k;
res.add(ans);
}
if(s >= target) {
s -= i;
i++;
} else {
j++;
s += j;
}
}
return res.toArray(new int[0][]);
}
复杂度分析:
时间复杂度 O(N): 其中 N = target ;连续整数序列至少有两个数字,而 i < j恒成立,因此至多循环 target 次( i, j都移动到target/2),考虑到解的稀疏性,将列表构建时间简化考虑为O(1);
空间复杂度 O(1) : 变量 i, j, s 使用常数大小的额外空间。
作者:jyd
链接:https://leetcode-cn.com/problems/he-wei-sde-lian-xu-zheng-shu-xu-lie-lcof/solution/jian-zhi-offer-57-ii-he-wei-s-de-lian-xu-t85z/