原题链接:和为S的连续正数序列_牛客题霸_牛客网
描述:
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列?
数据范围:0<n≤100
进阶:时间复杂度 O(n)
返回值描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
输入:
9
返回值:
[[2,3,4],[4,5]]
解题思路:
我们把时间复杂度就定位O(n),那么只能遍历一遍或者两遍。
解题思路是记录两个指针start和end,其中end>start。则分三种情况,
如果start和end之间的数字累加等于100,则list添加start和end之间的所有数字。
如果start和end之间的数字累加少于100,则end++,增加新的数字,继续循环。
如果start和end之间的数字累加大于100,则start++,去掉最前面的数字,继续循环。
代码:
public ArrayList<ArrayList<Integer>> FindContinuousSequence(int sum) {
ArrayList<ArrayList<Integer>> lists = new ArrayList<>();
int start = 0;
int end = 1;
int currentSum = start + end;
while (end <= (sum / 2) + 1) {
if (currentSum == sum) {
ArrayList<Integer> list = new ArrayList<>();
for (int i = start; i <= end; i++) {
list.add(i);
}
lists.add(list);
start = start + 2;
end++;
currentSum = currentSum - start * 2 + 3;
currentSum = currentSum + end;
} else if (currentSum > sum) {
currentSum -= start;
start++;
} else {
end++;
currentSum += end;
}
}
return lists;
}