刷题记录——剑指 Offer 57 - II. 和为 s 的连续正数序列

2021/9/28

题目描述

输入一个正整数 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

思路

根据target的值创建一个数组nums，范围[1, target/2+1]。使用双指针遍历数组，right=nums.length-1,left= right-1,让left逐渐向左移动。当nums[left]到nums[right]之和等于target时，将这个序列压入vector中，同时减小left和right；当nums[left]到nums[right]之和大于target时，需要同时减小left和right的值；当nums[left]到nums[right]之和只需要减小left的值。

代码

class Solution {
public:
    vector<vector<int>> findContinuousSequence(int target) {
        vector<vector<int>> v;
        vector<int> v_inner;
        if(target<1 || target>100000)
            return v;
        
        vector<int> nums;
        for(int i=1; i<= target/2 + 1; i++)
            nums.push_back(i);

        int right = nums.size() - 1;
        int left = right -1;
        while(left >= 0){
            v_inner = getInner(nums, left, right);    
            int sum =  getSum(nums, left, right);
            if(sum == target){
                sort(v_inner.begin(), v_inner.end());
                if(v_inner.size() >= 2)
                    v.push_back(v_inner); 
                left--;
                right--;               
            }else if(sum > target){
                left--;
                right--;
            }else{
                left--;
            }                            
        }
        sort(v.begin(), v.end());
        return v;
    }

    vector<int> getInner(vector<int> &nums, int left, int right){
        vector<int> v;
        while(left <= right){
            v.push_back(nums[left]);
            left++;
        }
        return v;
    }
    int getSum(vector<int> &nums, int left, int right){
        int sum = 0;
        while(left <= right){
            sum += nums[left];
            left++;
        }                    
        return sum;
    }
};

总结反思

本思路采用了枚举，代码的执行时间较长，耗费内存空间较大（求和的时候我竟忘了用高斯求和公式！！！）。官方题解使用的方法是滑动窗口。
设置左右指针，从开始位置维护一个子数组作为窗口，判断该窗口之和是否是target，若是则将结果加入；若小于则窗口右移；否则窗口左移。

class Solution {
public:
    vector<vector<int>> findContinuousSequence(int target) {
        vector<vector<int>> v;
        vector<int> v_inner;
        if(target<1 || target>100000)
            return v;
        
        int left = 1;
        int right = 2;
        while(left < right){
            int sum = (left+right)*(right-left+1)/2;
            if(sum == target){
                v_inner = getInner(left,right);
                v.push_back(v_inner);
                left++;
            }else if(sum < target){
                right++;
            }else{
                left++;
            }
        }
        sort(v.begin(),v.end());
        return v;
    }
    vector<int> getInner( int left, int right){
        vector<int> v;
        while(left <= right){
            v.push_back(left);
            left++;
        }
        return v;
    }
};

用时短！！！耗费内存空间少！！！