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<Integer,Integer> 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<Integer,Integer> 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;
}
}