单调栈原理+练习

发布于:2024-06-02 ⋅ 阅读:(138) ⋅ 点赞:(0)

首先用一道题引出单调栈

码蹄集 (matiji.net)

首先画一个图演示山的情况:

最暴力的做法自然是O(n方)的双循环遍历,这么做的思想是求出当前山右侧有多少座比它小的山,遇见第一个高度大于等于它的就停止。

但是对于我们所求的答案数,可以换一个想法,即对于一座山,求出它左侧有多少座山可以看到它。

对于5;7可以看到它。

对于3;7,5可以看到它。

对于4;7,5可以看到它。

可以发现,即求左侧有多少比当前值大的山的个数,那么我们可以维护一个数据结构,使得每次加进来一个数,都保持递减的顺序,然后每次加入前,都计算它的size大小,进行累加。比如:

一开始7进入;

然后5准备进入,进入前检查是否进入后会破坏单调性,发现不会,那么将当前的size累加到sum里,sum当前为1。5进入,现在数据结构为7,5;

然后3准备进入,检查发现不会破坏单调性,sum累加当前size(2),sum现在为(3)。3进入,现在数据结构为7,5,3;

然后4准备进入,检查发现会破坏单调性,就把比4小的全部扔出,于是3出去了。这时候sum累加当前size(2),sum现在为(5),然后4进入,数据结构现在为7,5,4;

结果也符合手算结果。

发现此过程进和出都是在同一侧进行,那么最适合的数据结构自然是栈了。这样的思想就是单调栈。

以下是代码:

#include<iostream>
#include<stack>
#define int long long
using namespace std;
signed main() {
    int n, num, sum=0;//n为山脉个数;num为每次读入的山脉高度,sum为总和
    stack<int> stk;//栈,我们要使它单调
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> num;
        //下面的while循环即用来维护单调性,本题要严格单调递减
        //每次检查栈顶元素是否小于等于当前高度,如果是则弹出,直到为空或者栈顶高度大于当前高度
        while (!stk.empty() && stk.top() <= num)
            stk.pop();
        //总和累加,stk.size即左侧比当前山脉高的山脉,也即能看到当前山的个数
        sum += stk.size();
        //入栈,栈此时满足单调性
        stk.push(num);
    }
    cout << sum;
}

(因为数据范围的要求,得用long long)

下面还有几道练习题

码蹄集 (matiji.net)

这道题需要我们脑补出究竟是哪种情况可以使海报数变少。

最多的情况自然是墙的个数n,考虑下面的情况:

如果出现这种情况,即低-高-低,并且左侧和右侧低的墙高度相同,那么就可以减少一张海报了。

注意中间高的墙可以不止一道,只要是被两个同高且相对低的墙夹住,整体就可以减少一张海报。

同样,可以没有所谓的高墙,即两道相连的墙同高,那自然也可以减少一张海报数。

而如果是这种情况:(涂橙色的代表墙)

那么即使左右两侧高度相同,还是需要用三张海报,因为题目要求海报不能超出边界。

我们同样维护一个数据结构,如果新输入的墙高于当前墙,那么就可以加入;而如果小于当前墙,那么就需要检查是否存在低-高-低的情况了。低的墙的高度即取决于新输入的墙的高度,将数据结构里所有比低墙要高的墙都弹出;若遇到同高的墙,则代表海报数可以减1,同时也弹出在数据结构里的那个同高的墙;若遇到比我们选定的低墙还低的墙,就结束弹出操作,并把新输入的墙加入。

以下面的数据为例,数字代表墙的高度:

2,2,3,4,2;以下是数据结构内容的变化

  • 2(第一个2加入)
  • 2(读入第二个2,发现同高,把前一个弹出,同时记录可以减1个海报)
  • 2-3(读入3,直接加入)
  • 2-3-4(读入4,直接加入)
  • 2(读入最后一个2,发现比顶部的墙小,开始弹出,发现4,3,2都要弹出,并且记录可以减1个海报数)

最后发现可以减少两个海报,那么就需要5-2=3个海报,就可以覆盖上述例子。

最终我们发现,维护的数据结构最终依然是单调的,并且操作在同侧进行,所以依然是单调栈的做法。

代码如下,代码量是很少的,但是需要知道其中的思想

#include<bits/stdc++.h> 
using namespace std;
int n,ans,d,w;
stack<int> st;
int main( )
{
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>d>>w;
        //只要栈顶的墙的高度大于等于墙的高度,都需要弹出
        while(!st.empty()&&w<=st.top()){
            if(st.top()==w)//如果遇见同高的墙,记录可以减少一个海报数
                ans++;
            st.pop();//弹出
        }
        st.push(w);//此时栈里的墙都比当前墙矮(或者栈为空),此时可以加入,就不会破坏单调性
    }
    cout<<n-ans;//最终结果为n-ans
    return 0;
}

下一道题

码蹄集 (matiji.net)

以样例为例,这道题也可用单调栈完成

维护一个单调递增的栈,一旦新增系数不满足单调性,即代表它是第一个小于栈顶元素系数的值。

按步骤拆解:

一开始1,3入栈

之后读入-3,开始检查,发现-3小于栈顶元素,于是x三次方的系数就变成-3,然后弹出3;之后-3与栈顶1对比,发现-3小于栈顶元素,于是x四次方的系数就是-3,然后弹出1 。-3入栈。

6入栈

读入1,比6小,所以x的系数是1 。但是比下一个栈顶元素-3大,于是入栈,现在栈里有-3,1;

最后x的平方和x的0次方系数为0.

这里有个问题,就是如果栈里存放系数的话,经过弹栈入栈,就判断不了这个系数是属于哪个x次幂的了,所以我们栈里改为存放下标i,通过下标读入存放在数组的系数。

以下是代码:

#include<bits/stdc++.h> 
#define ll long long
const int MOD=99887765;
const int N=5e6+7;
using namespace std;
stack<int> st;
//a是原系数数组,b是新的系数数组
int n,x,a[N],b[N];
ll ans;
int main( )
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>x;
    for(int i=1;i<=n+1;i++){
        cin>>a[i];
        //如果读入的系数小于栈顶的系数(注意栈里存放的是下标,于是通过a[st.top()]来获得栈顶的系数)
        while(!st.empty()&&a[i]<a[st.top()]){
            b[st.top()]=a[i];//栈顶位置的x次幂的系数设置为当前读入的系数(是第一个小于原系数的值)
            //弹栈
            st.pop();
        }
        //注意如入栈的是下标
        st.push(i);
    }
    //其实最好是显式初始化一下b数组,把初始值都设为0,虽然不这样做默认初值也是0.这样没处理的下标的保存的新系数自动设为0
    //这里就是计算多项式了
    for(int i=1;i<=n+1;i++){
        ans=ans*x+b[i];
        ans=(ans%MOD+MOD)%MOD;
    }
    cout<<ans;
    return 0;
}

以上就是一些单调栈的简单示例了。

似乎可以总结出一个模板了:

1.循环读入新元素

2.用一个while循环,通过对比栈顶和新元素,来把不满足单调性的栈顶弹出

3.新元素压栈,保持单调性

至于题目有什么要求,就要在模板里进行一些其他操作了


网站公告

今日签到

点亮在社区的每一天
去签到