【剑指offer中等部分14】和为S的连续正数序列（java）

最新推荐文章于 2024-10-26 18:58:40 发布

原创最新推荐文章于 2024-10-26 18:58:40 发布 · 191 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

剑指offer（java）

本文介绍了一种寻找所有连续正整数序列的方法，这些序列的和等于给定数值S。通过使用两个指针技巧，实现了有效遍历并记录符合条件的序列。

题目描述

小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
在这里插入图片描述

分析

我们可以定义两个指针low和high，从1，2数字开始往后找，也就是进行穷举。如果找到与s相等的，则将该序列添加进入集合中。若当前窗口内的值之和小于sum，那么右边指针high往右移一下。若当前窗口内的值之和大于sum，那么左边指针low右移一下。
注意：题中说的是连续序列

代码如下：

import java.util.ArrayList;
public class Solution {
    public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
        //返回的是集合嵌套的形式，外面集合装着里面的连续数列，该数列之和等于S
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        int low = 1, high = 2; //从数字1，2开始往后滑动
        while(high > low){
            int temp = (high + low)*(high - low + 1) / 2;    //连续的，差为1的一个序列，那么求和公式是(a0+an)*n/2
            //若相等，则将序列添加进入集合
            if(temp == sum){
                ArrayList<Integer> list = new ArrayList<>();
                for(int i = low ; i <= high ; i++){
                    list.add(i);
                }
                result.add(list);
                low++;//往后走
            }
            //若当前窗口内的值之和小于sum，那么右边窗口右移一下
            else if(temp < sum){
                high++;
            }else{
            //若当前窗口内的值之和大于sum，那么左边窗口右移一下
                low++;
            }
        }
        return result;
    }
}