Leetcode.829 连续整数求和

感觉画质不如…原神

于 2023-03-09 19:44:51 发布

阅读量362

点赞数 1

分类专栏： Leetcode 文章标签：算法 leetcode 等差数列

本文链接：https://blog.csdn.net/m0_74396439/article/details/129429305

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

文章介绍了LeetCode第829题的解题思路，涉及寻找连续正整数之和等于给定值n的所有组合。通过分析等差数列的性质，得出求解公式并给出C++和Java的代码实现，主要利用了平方根下界和偶数特性来优化搜索过程。

摘要由CSDN通过智能技术生成

题目链接

Leetcode.829 连续整数求和 Rating ： 1694

题目描述

给定一个正整数 n，返回连续正整数满足所有数字之和为 n的组数。

示例 1:

输入: n = 5
输出: 2
解释: 5 = 2 + 3，共有两组连续整数([5],[2,3])求和后为 5。

示例 2:

输入: n = 9
输出: 3
解释: 9 = 4 + 5 = 2 + 3 + 4

示例 3:

输入: n = 15
输出: 4
解释: 15 = 8 + 7 = 4 + 5 + 6 = 1 + 2 + 3 + 4 + 5

提示：

$1 <= n <= 10^9$

分析：等差数列

按照题意，我们要求的 n其实是一段 d = 1的等差数列的和。

等差数列 $a_n$ 的通项为 $a_n = a_1 + (n-1)*d$

等差数列的求和公式为 $\frac{(a_1+a_n)*n}{2}$

我们假设选取连续的一段 k个元素是满足要求的，即 $\frac{(a_1+a_k)*k}{2} = n$ ，这里的 $a_1$ 只是我们假设的第一项。

证明：

先化简为 $2a_1 + (k-1) * d = \frac{2n}{k}$

因为 d = 1，所以 $2a_1 + k-1 = \frac{2n}{k}$ ，在整理一下 $\frac{2n}{k} - k + 1 = 2a_1$ 。

因为我们是从 1到n中开始选的，所以 $a_1\geq1$ ，即 $\frac{2n}{k} - k + 1 \geq 2$ ， $\frac{2n}{k} \geq k + 1$ 。

所以 $\frac{2n}{k} > k$ ， $\sqrt{2n}$ ，最后求出了 $k$ 的最大值。

因为 $2a_1 + k-1 = \frac{2n}{k}$ ， $2a_1 + k-1$ 肯定是一个正整数，所以 $2 n$ 就是 $k$ 的倍数。

又因为 $\frac{2n}{k} - k + 1 = 2a_1$ ， $2a_1$ 是一个偶数，所以 $\frac{2n}{k} - k + 1$ 就是 2的倍数。

所以我们最终只需要从 $1$ 遍历到 $\sqrt{2n}$ ，满足这两个条件的就是一个答案。

时间复杂度： $O(\sqrt{n})$

C++代码：


class Solution {
public:
    int consecutiveNumbersSum(int n) {
        int ans = 0;
        for(int k = 1;k * k <= 2 * n;k++){
            if((2*n) % k == 0 && ((2*n)/k - k + 1) % 2 == 0) ans++; 
        }
        return ans;
    }
};

Java代码：


class Solution {
    public int consecutiveNumbersSum(int n) {
        int ans = 0;
        for(int k = 1;k * k <= 2 * n;k++){
            if((2*n) % k == 0 && ((2*n)/k - k + 1) % 2 == 0) ans++; 
        }
        return ans;
    }
}