剑指offer -- 和为s的连续正数序列

leetcode链接:https://leetcode-cn.com/problems/he-wei-sde-lian-xu-zheng-shu-xu-lie-lcof/

输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。

序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。

示例 1:

输入:target = 9
输出:[[2,3,4],[4,5]]

示例 2:

输入:target = 15
输出:[[1,2,3,4,5],[4,5,6],[7,8]]

解题思路:由于是连续的正数序列,因此可以使用公式:(首项+尾项)* 项数 / 2 来求和,首先计算出数列最大有可能的尾项,然后首项从1开始进行遍历,若此序列之和大于target,则首相后移,若符合条件,则将首尾项存入map,若小于target,则退出当前循环,尾项减一,继续寻找。最后将map中符合条件的序列输出为数组。

class Solution {
    public int[][] findContinuousSequence(int target) {
        Map map = new TreeMap<>();
        int end = target/2+1;
        int count = 0;
        while(end > 0){
            int start = 1;
            while(start != end){
                int sum = (start+end) * (end-start+1) / 2;
                if(sum==target){
                    map.put(start,end);
                    count++;
                    break;
                }else if(sum>target){
                    start++;
                }else{
                    break;
                }
            }
            end--;
        }

        int[][] result = new int[count][];
        int i=0;
        for(Map.Entry tmp : map.entrySet()){
            int start1 = tmp.getKey();
            int end1 = tmp.getValue();
            int j=0;
            int[] sub = new int[end1-start1+1];
            while(start1 <= end1){
                sub[j++] = start1;
                start1++;
            }
            result[i] = sub;
            i++;
        }
        return result;
    }
}

 

你可能感兴趣的:(剑指offer系列)