LeetCode 5958. 股票平滑下跌阶段的数目(滑动窗口)

文章目录


1. 题目

2. 解题


1. 题目


给你一个整数数组 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 <= 10^5
1 <= prices[i] <= 10^5



2. 解题


  • 滑动窗口 [i, j] 记录连续下降 1 的区间,每次 加入答案 j-i+1
  • 不满足连续下降 1,则重新开始记录区间
class Solution {
public:
    long long getDescentPeriods(vector<int>& prices) {
        long long n = prices.size(), i = 0, j = 1, ans = 1;
        while(j < n)
        {
            while(j < n && prices[j]+1 == prices[j-1])
            {
                ans += j-i+1;
                j++;
            }
            i = j;
            if(i < n)
                ans++;
            j++;
        }
        return ans;
    }
};

120 ms 90.8 MB C++

上一篇:顺序表


下一篇:LeetCode 5957. 向字符串添加空格