剑指offer-和为S的连续正数序列
题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
题解:
使用map,记录sum值,从小到大的push_back到ans中。
class Solution { public: vector<vector<int> > FindContinuousSequence(int sum) { vector<vector<int> > ans; map<int, int> mp; int t = 0, tmp =0; mp[0] = 0; for(int i=1; i<sum; ++i){ t += i; mp[t] = i; tmp = t - sum; if( mp.find( tmp ) != mp.end()){ vector<int> a; for(int j=mp[tmp]+1; j<=i; ++j){ a.push_back(j); } ans.push_back(a); } } return ans; } };