劍指Offer - 和為S的連續正數序列(Java實現)
阿新 • • 發佈:2019-01-02
題目描述:
小明很喜歡數學,有一天他在做數學作業時,要求計算出9~16的和,他馬上就寫出了正確答案是100。但是他並不滿足於此,他在想究竟有多少種連續的正數序列的和為100(至少包括兩個數)。沒多久,他就得到另一組連續正數和為100的序列:18,19,20,21,22。現在把問題交給你,你能不能也很快的找出所有和為S的連續正數序列?
Good Luck!
輸出描述:
輸出所有和為S的連續正數序列。序列內按照從小至大的順序,序列間按照開始數字從小到大的順序
思路分析:
本題的關鍵點是想到雙指標,利用雙指標形成滑動視窗。
1,獲取滑動視窗中數的總和,與已知sum進行比較。
2,當窗口裡數累和大於sum時,將左指標向右移動一位,繼續2。
3,若窗口裡數累和小於sum時,將右指標向右移動一位,繼續2。
4,若相等時,將滑動視窗中的數存入ArrayList集合中,並將該集合存入ArrayList<ArrayList>集合中,左指標移動一位,繼續2。
注意,所有的迴圈都是基於左指標小於右指標這個條件。若不滿足條件直接返回。
程式碼實現如下:
mport java.util.ArrayList; //雙指標的思路 public class Solution { public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) { int low = 1; int high = 2; ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>(); while(low < high){ int getsum = (low + high)*(high-low+1)/2; if(getsum < sum){ high++; }else if(getsum == sum){ ArrayList<Integer> list = new ArrayList<Integer>(); for(int i = low ; i <= high; i++){ list.add(i); } result.add(list); low++; }else{ low++; } } return result; } }