题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
【C++解法】
1.暴力求解
1.1 双指针滑动窗口+状态转移方程迭代+累加求和
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
int l = 1, r = 1, sumx = 1, mid = (sum+1)/2;
vector<vector<int> > ans;
while(l <= mid){
r++;
sumx += r;
while(sumx > sum){//这个循环的隐患在于循环结束时,beign==end而且和等于sum
sumx -= l;
l++;
}
if(sumx == sum && l != r){//防止出现某个数组只有1项的情况
vector<int> tmp;
for(int i = l;i <= r;i ++) tmp.push_back(i);
ans.push_back(tmp);
}
}
return ans;
}
};
1.2 标准双指针+求和公式
时间复杂度:O(N)
空间复杂度:O(1)
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > results;
int low = 1, high = 2;
while (high > low) { //终止条件,因为题干要求至少2个数
int sumx = (high + low) * (high - low + 1) / 2; //求和公式
if (sumx < sum) {high++;} //当前窗口内的值之和小于sum,右框右移一下
else if (sumx == sum) { //相等,就将窗口内所有数添加进结果集
vector<int> res;
for (int i = low; i <= high; i++) {res.push_back(i);}
results.push_back(res);
low++; //缩小框,左边前移
} else {low++;} //如果当前窗口内的值之和大于sum,左窗框左移一下
}
return results;
}
};
2.数学分析
- n = 2k + 1时,n项连续正数序列的和为S的条件: n & 1 && S / n == 0 解读 逻辑与的左边要求n为奇数,右边要求整个序列的平均数恰好为中间数。
- n = 2k时,n项连续正数序列的和为S的条件: S % n * 2 == n 解读 S % n 的结果是中间两项左边的那项,乘2刚好是项数。举例,现有S = 39,6个连续正数序列和式能不能为S呢?套用公式,39 % 6 * 2 =6 == 6,我们也知道,这次的序列是 4、5、6、7、8、9,取余的结果为3对应着值为6的那一项,也就是中间项左边的那一项。
- 和为S,项数为n,如何写出这个序列? S / n - (n-1) / 2 解读 执行的除法是地板除法(floor),不管最终结果有无小数都直接舍去。仍使用上述例子,39 / 6 = 6,6恰好是中间项左边的那一项,6 - (6-1)/ 2 = 4,恰好是序列最左端。序列写出来就没问题。
- 等差序列公式S = (a0 + a0 +n-1)* n / 2进行放缩,就有S > n平方 / 2;即n < 根号2S。这样做的话可以减少遍历次数。
可以先求首项
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > allRes;
for (int n = sqrt(2 * sum); n >= 2; --n) {
if (((n&1)==1 && sum%n == 0) || (sum%n*2==n)){
vector<int> res; //j用于计数,k用于遍历求值
int k = sum/n-(n-1)/2;
for(int j=0;j<n;j++) {
res.push_back(k);
k++;
}
allRes.push_back(res);
}
}
return allRes;
}
};
也可以判断首项是否整数
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > allRes;
if(sum < 3) return allRes;
for(int k = sqrt(2 * sum); k >= 2; --k) { //k为项数
double cheat = (2.0*sum/k - k + 1)/2.0; //为了验证首项是否为整数
int begin = cheat;
if(begin == cheat) { //首项确实为整数
vector<int> res;
for(int i = begin; i < begin + k; ++i) res.push_back(i);
allRes.push_back(res);
}
}
return allRes;
}
};
3.换元法
(a + b)(b - a + 1) = 2 * sum;此为等差数列公式。
令i = b - a + 1(项数), j = a + b(首末项之和);现讨论取值范围。i >= 2(序列中至少有2项), j >= 3(序列之和至少为3);隐藏的关系是: j > i同时还有 i * j = 2 * sum,进行放缩之后就有 i * i < 2 * sum,即 i < 根号(2 * sum)。对i进行遍历,找出i,j∈正整数且j - i + 1为偶的取值。
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > allRes;
if(sum < 3) return allRes;
int twofoldSum = 2 * sum, i, j;
for(i = sqrt(twofoldSum); i >= 2; --i) { //枚举出合适的项数
if(twofoldSum % i == 0) { //首末项之和为整数
j = twofoldSum/i; //求出首末项之和
int temp = j-i+1; //解方程得到2倍的左值等于j-i+1;要求左边界为整数
if (!(temp&1)) { //要求temp是偶数
int begin = temp>>1;
int end = j-begin; //j的实际意义是首末项之和
vector<int> res;
for(int k = begin; k <= end; ++k) res.push_back(k);
allRes.push_back(res);
}
}
}
return allRes;
}
};
【C解法】
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param sum int整型
* @return int整型二维数组
* @return int* returnSize 返回数组行数
* @return int** returnColumnSizes 返回数组列数
*
* C语言声明定义全局变量请加上static,防止重复定义
*/
int** FindContinuousSequence(int sum, int* returnSize, int** returnColumnSizes ) {
int** results = malloc(sizeof(int*)*1);
(*returnColumnSizes) = malloc(sizeof(int)*1);
int low = 1, high = 2;
*returnSize = 0;
while (high > low) {
int sumx = (high + low) * (high - low + 1) / 2;
if (sumx < sum) {high++;}
else if (sumx == sum) {
(*returnColumnSizes)[*returnSize] = high - low + 1;
results[*returnSize] = malloc(sizeof(int) * ((*returnColumnSizes)[*returnSize]));
for (int i = low; i <= high; i++) {results[*returnSize][i - low] = i;}
(*returnSize)++;
low++;
} else {low++;}
}
return results;
}
【Java解法】
import java.util.ArrayList;
public class Solution {
public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
ArrayList<ArrayList<Integer> > results = new ArrayList<>();
int low = 1, high = 2;
while (high > low) {
int sumx = (high + low) * (high - low + 1) / 2;
if (sumx < sum) {high++;}
else if (sumx == sum) {
ArrayList<Integer> res = new ArrayList<>();
for (int i = low; i <= high; i++) {res.add(i);}
results.add(res);
low++;
} else {low++;}
}
return results;
}
}
【Kotlin解法】
object Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
* @param sum int整型
* @return int整型二维数组
*/
fun FindContinuousSequence(sum: Int): Array<IntArray> {
var results = ArrayList<IntArray>()
var low : Int = 1
var high : Int = 2
while (high > low) {
var sumx : Int = (high + low) * (high - low + 1) / 2;
if (sumx < sum) {high++}
else if (sumx == sum) {
results.add(IntArray(high - low + 1){it + low})
low++
} else {low++}
}
return results.toTypedArray()
}
}
【Rust解法】
struct Solution{
}
impl Solution {
fn new() -> Self {
Solution{}
}
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
* @param sum int整型
* @return int整型二维数组
*/
pub fn FindContinuousSequence(&self, sum: i32) -> Vec<Vec<i32>> {
let mut result : Vec<Vec<i32>> = Vec::new();
let mut low = 1;
let mut high = 2;
while high > low {
let sumx = (high + low) * (high - low + 1) / 2;
if sumx < sum {high += 1;}
else if sumx == sum {
let mut res : Vec<i32> = Vec::new();
for i in low..=high {res.push(i);}
result.push(res);
low += 1;
} else {low += 1;}
}
result
}
}
参考文献: