LeetCode 2110.股票平滑下跌阶段的数目

吃着火锅x唱着歌2025-09-15 10:27

给你一个整数数组 prices ，表示一支股票的历史每日股价，其中 prices[i] 是这支股票第 i 天的价格。

一个平滑下降的阶段定义为：对于连续一天或者多天，每日股价都比前一日股价恰好少 1 ，这个阶段第一天的股价没有限制。

请你返回平滑下降阶段的数目。

示例 1：

输入：prices = [3,2,1,4]

输出：7

解释：总共有 7 个平滑下降阶段：

3\], \[2\], \[1\], \[4\], \[3,2\], \[2,1\] 和 \[3,2,1

注意，仅一天按照定义也是平滑下降阶段。

示例 2：

输入：prices = [8,6,7,7]

输出：4

解释：总共有 4 个连续平滑下降阶段：[8], [6], [7] 和 [7]

由于 8 - 6 ≠ 1 ，所以 [8,6] 不是平滑下降阶段。

示例 3：

输入：prices = [1]

输出：1

解释：总共有 1 个平滑下降阶段：[1]

提示：

1 <= prices.length <= 105^55

1 <= prices[i] <= 105^55

分组循环，每一组都是平滑下跌阶段，找出全部阶段：

cpp 复制代码

class Solution {
public:
    long long getDescentPeriods(vector<int>& prices) {
        int n = prices.size();

        long long ans = 0;

        int i = 0;
        while (i < n) {
            int start = i;
            ++i;
            while (i < n && prices[i] + 1 == prices[i - 1]) {
                ++i;
            }

            int day = i - start;
            ans += (long long)day * (day + 1) / 2;
        }

        return ans;
    }
};

如果prices的长度为n，则此算法时间复杂度为O(n)，空间复杂度为O(1)。