劍指offer -- 和為 S 的連續正數序列
阿新 • • 發佈:2018-11-05
題目描述
小明很喜歡數學,有一天他在做數學作業時,要求計算出9~16的和,他馬上就寫出了正確答案是100。但是他並不滿足於此,他在想究竟有多少種連續的正數序列的和為100(至少包括兩個數)。沒多久,他就得到另一組連續正數和為100的序列:18,19,20,21,22。現在把問題交給你,你能不能也很快的找出所有和為S的連續正數序列? Good Luck!
- 輸出描述
輸出所有和為S的連續正數序列。序列內按照從小至大的順序,序列間按照開始數字從小到大的順序
AC程式碼
import java.util.ArrayList;
public class Solution {
public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
ArrayList<ArrayList<Integer>> resList = new ArrayList<>();
if(sum < 3) return resList;
int p1 = 1;
int p2 = 2;
int curSum = 3;
while(p2 <= (sum + 1) / 2 ){
if(curSum == sum){
addList(p1, p2, resList);
curSum = curSum - p1 + p2 + 1;
p1++;
p2++;
}
else if(curSum < sum){
p2++;
curSum += p2;
}
else {
curSum -= p1;
p1++;
}
}
return resList;
}
private void addList(int p1, int p2, ArrayList<ArrayList<Integer>> resList){
ArrayList<Integer> list = new ArrayList<>();
for(int i = p1;i <= p2;i++) list.add(i);
resList.add(list);
}
}