单调栈板子

单调栈

用于求解数组每个位置上左边/右边离自己最近的且严格小于/大于自己位置上的数位置
时间复杂度O(N)(每个元素下标进栈一次出栈一次)
元素下标能表示的含义比元素本身要多
(ps:注意数组长度,过大就要开到全局变量中,否则异常退出orz)

方法(求每个位置上离自己最近且严格小于自己的元素位置(严格大压小)),允许有重复值

int ans[maxn][2];//用来存放每个位置上左答案和右答案
int st[maxn];//用数组模拟栈
void compute(int arr[])
{
    int r=0;
    int cur;
    //遍历阶段
    for(int i=0;i<n;i++)
    {
        while(r>0&&arr[st[r-1]]>=arr[i])//栈顶元素大于或等于要进栈的元素
        {
            cur=st[--r];//栈顶元素的index
            ans[cur][0]= r>0 ? st[r-1]:-1;(如果下面没压下标,就返回-1当作不存在)
            ans[cur][1]=i;(如果栈顶元素大于要进栈的元素,此答案为真。否则,先记录下来,再修正)
        }
        st[r++]=i;//元素下标进栈
    }
    //清算阶段
    while(r>0)//当栈里还有元素下标时
    {
        cur=st[--r];
        ans[cur][0]= r>0?st[r-1]:-1;
        ans[cur][1]=-1;//右边没有
    }
    //修正阶段
    for(int i=n-2;i>=0;i--)//从右往左修正(n-1位置元素右侧一定是-1,无需验证)
    {
        if(ans[i][1]!=-1&&arr[ans[i][1]]==arr[i])//只需验证右答案正确性(如果i位置元素==右答案记录的元素,就把i位置右答案“右移动”)
        {
            ans[i][1]=ans[ans[i][1]][1];
        }
    }
}
posted @   Marinaco  阅读(7)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· C#/.NET/.NET Core优秀项目和框架2025年2月简报
· Manus爆火,是硬核还是营销?
· 一文读懂知识蒸馏
· 终于写完轮子一部分:tcp代理 了,记录一下
点击右上角即可分享
微信分享提示