leetcode 3578

3578: 统计极差最大为K的分割方式数

思路:单调队列+动态规划

单调队列是一种特殊的队列数据结构,它维护队列中的元素具有单调性(单调递增或单调递减)。它主要用于解决++滑动窗口类问题++ ,可以在O(n)时间内解决一些看似需要O(n²)时间的问题,适合处理需要维护区间最值的问题。

单调队列的核心思想是:

  • 维护一个双端队列(deque)

  • 队列中的元素保持单调性

  • 及时移除不再有用的元素

  • 保证队列头部始终是当前窗口的最值

0 <= k <= 10^9

一般定义 fi+1 表示++前缀++ nums0 到 numsi 在题目约束下,分割出的最少(最多)子数组个数,本题是定义成分割方案数。

动态规划:我们设 dpi+1 表示以 i 为结尾的前缀子数组 nums0⋯i 满足题目要求的分割方案数,++初始化 dp0=1,此时前缀为空也看作为一种合法的分割。++ 可分割成一个或多个 **非空** 的连续子段枚举最后一个子数组的左端点 j,那么问题变成前缀 nums0 到 numsj−1 在题目约束下的分割方案数,即 fj。当子数组++右端点 i 固定时++ ,由于子数组越长,最大值越大,最小值越小,最大最小的差值越可能大于 k。所以符合要求的++左端点 j++ 一定在一个连续区间 L,i 中。累加 fj递归,分割方案数求和

答案为 fn

复制代码
deque<int> min_q, max_q;

两个完全独立的双端队列,同步地 维护一个滑动窗口的最小值最大值(即"单调队列"最经典的用法)

复制代码
while(!min_q.empty() && x<=nums[min_q.back()]){ 
    min_q.pop_back();
}
min_q.push_back(i);
while(!max_q.empty() && x>=nums[max_q.back()]){
    max_q.pop_back();
}
max_q.push_back(i);
  • min_q:把"大于等于 x 的"从尾端踢掉 → 队列里严格单调递增(头小尾大)。

  • max_q:把"小于等于 x 的"从尾端踢掉 → 队列里严格单调递减(头大尾小)。

所以
min_q递增队列 ,队头存当前窗口最小值
max_q递减队列 ,队头存当前窗口最大值

复制代码
class Solution {
public:
    int countPartitions(vector<int>& nums, int k) {
        constexpr int MOD=1'000'000'007;
        int n=nums.size();
        deque<int> min_q,max_q; //获得窗口中的最大、最小值
        vector<int> f(n+1); //动态规划,f[0]~f[n]递归
        f[0]=1;
        long long sum_f=0; // 窗口中的 f[i] 之和
        int left=0; 

        for(int i=0;i<n;i++){
            int x=nums[i];
            sum_f+=f[i]; //滑动窗口-入

            while(!min_q.empty() && x<=nums[min_q.back()]){ 
                min_q.pop_back();
            }
            min_q.push_back(i);
            while(!max_q.empty() && x>=nums[max_q.back()]){
                max_q.pop_back();
            }
            max_q.push_back(i);

            while(nums[max_q.front()]-nums[min_q.front()]>k){ //最大值与最小值作差超过k
                sum_f-=f[left]; //滑动窗口-出,类前缀和
                left++;
                if(min_q.front()<left){
                    min_q.pop_front();
                }
                if(max_q.front()<left){
                    max_q.pop_front();
                }
            }
            f[i+1]=sum_f%MOD;
        }

        return f[n];
    }
};
相关推荐
爱编程的小新☆几秒前
【LeetCode】从递归到 Flood Fill:5 道题吃透 DFS 的选择、回溯与标记
java·算法·leetcode·深度优先·回溯·flood fill
Water_Sunzhipeng5 分钟前
2024牛客暑期多校训练营1
算法
hetao17338375 分钟前
2026-08-09~08-14 hetao1733837 的刷题记录
c++·算法
技术小黑8 分钟前
RNN算法实战系列06 | LSTM 实现糖尿病探索与预测
rnn·算法·lstm
evans在进步10 分钟前
LeetCode 33:搜索旋转排序数组——Java 两阶段二分查找详解
java·python·leetcode
疯狂打码的少年29 分钟前
【数据结构】二叉树的性质(五大性质+计算)
数据结构·笔记·算法
wabs6661 小时前
关于图论【A*算法 | 卡码网127.骑士的攻击的思考】
数据结构·算法·图论·卡码网·广搜的改进版
CIO_Alliance1 小时前
AI认知系列(3)| 数据、算法、算力、场景四要素协同
人工智能·算法·ipaas·系统集成·企业cio联盟·企业级ai化转型
凉茶钱1 小时前
【数据结构】堆的应用
c语言·数据结构
Forever Nore1 小时前
LeetCode 13 罗马数字转整数 - 按规则处理
linux·服务器·leetcode