题目链接:https://leetcode-cn.com/problems/consecutive-numbers-sum/description/
给定一个正整数 N
,试求有多少组连续正整数满足所有数字之和为 N
?
示例 1:
输入: 5
输出: 2
解释: 5 = 5 = 2 + 3,共有两组连续整数([5],[2,3])求和后为 5。
示例 2:
输入: 9
输出: 3
解释: 9 = 9 = 4 + 5 = 2 + 3 + 4
示例 3:
输入: 15
输出: 4
解释: 15 = 15 = 8 + 7 = 4 + 5 + 6 = 1 + 2 + 3 + 4 + 5
说明: 1 <= N <= 10 ^ 9
仔细观察我们会知道这是一道关于等差数列k项和的题。我们把数列长度k划为起始数start与终点数end的差,即k=end-start+1。等差数列k项和可以写成(start+end)*(end-start+1)=2*N。但是题目数据范围1e9,直接暴力匹配肯定是会超时的。所以我们可以将化为寻找2*N的因数的问题,也就是(start+end)和(end-start+1)必定是2*N的因子,直接将问题的时间复杂度化为O(logn)。
PS:设i=start+end,(start+end)*(end-start+1)=2*N => (end-start)*(start+end)+(start+end)=2*N => end=(2*N/i+i-1)/2,start=end-2*N/i+1.
代码如下:
int consecutiveNumbersSum(int N) {
int num=1;
int n=2*N;
for(int i=2;i<=sqrt(n);i++){
if(n%i==0){
int end=(n/i+i-1)/2;
int start=end-n/i+1;
if((start+end)*(end-start+1)==n)
num++;
}
}
return num;
}