小白也能看懂的算法笔记:给定一个数组序列, 需要求选出一个区间, 使得该区间是所有区间中经过如下计算的值最大的一个(单调栈)

问题描述

题目如下:

给定一个数组序列, 需要求选出一个区间, 使得该区间是所有区间中经过如下计算的值最大的一个:
区间中的最小数 * 区间所有数的和最后程序输出经过计算后的最大值即可,不需要输出具体的区间。如给定序列 [6 2 1]则根据上述公式, 可得到所有可以选定各个区间的计算值:
从上述计算可见选定区间 [6] ,计算值为 36, 则程序输出为 36。
区间内的所有数字都在[0, 100]的范围内;
[6] = 6 * 6 = 36;
[2] = 2 * 2 = 4;
[1] = 1 * 1 = 1;
[6,2] = 2 * 8 = 16;
[2,1] = 1 * 3 = 3;
[6, 2, 1] = 1 * 9 = 9;


这篇文章会介绍两种解题思路,第一种解题思路看似简单直观,但却为第二种思路进行了铺垫。这是非常重要的。

第一种方法:穷举法

第一个思路是最容易想到的方法。我们把每个元素都看作最小值,从它向左右两边延伸划定一个集合,集合中包括的其他元素当然要比这个最小值大,且集合的范围越大越好。

遍历集合中的每一个元素,最后得到最大值。

打个比方,集合为1、3、4、2、6、5,我们找到每个数的子集,该子集满足三个条件:

  1. 该子集中所有元素都相邻。
  2. 这个数必须是子集中最小的元素。
  3. 该子集中元素要尽量多。

好了现在为每个元素都找到了这样的子集:

1: {1, 3, 4, 2, 6, 5}
3: {3, 4}
4: {4}
2: {3, 4, 2, 6, 5}
6: {6}
5: {6, 5}

那么我们是怎样找到这样的子集呢?很简单,就是从当前元素开始向左遍历,遇到\color{blue}{第一个比它小的元素}停止,这样就找到子集的\color{blue}{左边界}。然后再向右遍历,遇到\color{green}{第一个比它小的元素}停止,这样就找到子集的\color{green}{右边界}。得到了左右边界,就相当于我们确定了子集。

然后用最小值乘以集合内的元素之和,即
1\times(1+3+4+2+6+5)
3\times(3+4)
4\times4
......

找到最大的那个就可以了。

这种方法的时间复杂度是O(n^2),实际上是不能让全部数据AC的。下面我介绍第二种思路。

第二种思路:使用栈的思想

实际上我是学习了这位大佬的博客,大佬写得有些生涩,不适用于我这样的小白,所以我花了一晚上理解并记录下我对这种解法的二次理解。这种方法采用了单调栈的数据结构,时间复杂度为O(n),非常牛批。

1. 核心要点

我们依旧以集合1、3、4、2、6、5为例,事实上,理解方法二是一定要在理解方法一的基础上进行的。上文已经提过,找到每个元素对应的子集的核心就是找到子集的\color{blue}{左边界}\color{green}{右边界},而找到左右边界的核心就是找到左右第一个比当前元素小的元素。明白了这一点,我们就可以开始建栈了。

遍历集合中的元素,第一个元素为1,然后思考这样一个问题:\color{red}{现在的我能确定1对应的子集的左右边界吗?}显然,我们能确定左边界,因为1已经是最左边的元素了;但我们不能确定右边界,因为如果1的下一个元素比1要大,那么1的右边界就可以往右扩大一个。当前情况下我们不能得到1的子集,所以让1\color{red}{进栈}。然后遍历第二个元素。

第二个元素值为3。我们依旧思考这个问题:我们能确定下3对应的子集的左右边界吗?同样,我们能确定左边界,因为3的上一个元素是1,1比3小,所以左边界不能再扩张了,而1正是此时的栈顶元素;但右边界同样不能确定,因为我们不知道3的下一个数字是否会比3大。因为当下我们不能确定这个子集,所以3进栈。遍历第三个元素。

第三个元素为4,与3同理,左边界是栈顶元素3但不知道右边界。4进栈。


第四个元素是2,这时情况出现了改变——2比4要小。我们突然发现,我们可以确定元素4对应子集的右边界了!此时4对应的子集已经完全可以确定,因此让4\color{red}{出栈},并计算这个值也就是\color{blue}{4\times 4=16}\

现在栈顶元素是3,而2竟然比3也小!也就是说3对应子集的右边界也找到了。让3出栈。现在我们可以确定下3对应的子集是{3, 4},计算这个值\color{blue}{3\times(3+4)=21}

现在栈顶元素是1,2比1大,1的右边界依旧不能确定,因此1不出栈。2的左边界确定,正是当前的栈顶元素1,而右边界不能确定,2进栈。此时栈变为


注意:栈顶2下面的元素1正是2对应的子集的左边界

同理,6比2大,进栈。

5比6小,6的右边界确定,6出栈,并计算题目要求的值即\color{blue}{6\times 6=36}。此时栈顶元素为2,5比2大,5进栈。

到这里,题目中给出的元素已经全部遍历结束,但栈并不为空。为了让栈为空,我们再在结尾多加一个元素0。0与栈顶元素为5,0比5小,5的右边界确定,5出栈,计算题目中要求的值\color{blue}{5\times(5+6)=55}

然后比较0与栈顶元素2,0比2小,与上面同理,计算\color{blue}{2\times(3+4+2+6+5)=40},2出栈。\color{green}{(看到这里,我相信你已经产生了疑问,括号中的内容(3+4+2+6+5)是在怎么计算的呢?别着急,我们下面会继续讲这个问题)}

比较0与此时的栈顶元素1,0比1小,与上面同理,计算\color{blue}{1\times(1+3+4+2+6+5)=21}。1出栈。

循环结束。

以上就是这个算法,最核心的思想。总结一下,就是其核心要点为:

新建一个栈,然后逐个遍历集合中的元素,如果栈为空或该元素大于栈顶元素,则该元素入栈;否则让栈中的元素不停地出栈,直到栈顶元素小于该元素时,该元素入栈

2. 计算子集中的所有元素之和

这就是上文中绿字标注的问题,我们之前刻意回避了,蓝色的公式都涉及到了它的计算,现在我们来思考一下它们时怎么被计算出来的。

我们以元素2出栈时的情况为例,2对应的子集为\{\color{blue}3, \color{blue}4, \color{red}2, \color{green}6, \color{green}5\},其中3和4位于2的左侧,5和6位于2的右侧。集合中所有元素之和=当前元素+右侧元素之和+左侧元素之和。我们用temp_sum来表示。

右侧元素之和很好计算。2的右侧元素集合\left\{\color{green}6, \color{green}5\right\},正是5对应的元素集合,而2的出栈一定是紧跟在5出栈之后的,所以2右侧元素之和=5的temp_sum。

左侧元素之和没那么容易计算了。但事实上,左侧元素构成的集合\left\{\color{blue}3, \color{blue}4\right\}单独拿出来,正是元素3对应的集合,而这个集合元素之和在前面元素3出栈时计算过它的temp_sum了。只要我们当时把这个temp_sum保存下来,现在就可以直接使用了。

因此我们再多新建一个栈leftsum_stack。leftsum_stack的元素和前面的栈(为了区分我将前面用到的栈命名为stack)的元素是一一对应的,leftsum_stack中的某个元素代表stack中对应位置的元素的左边元素之和。stack有元素进栈时leftsum_stack也进栈,stack有元素出栈时leftsum_stack也出栈。

因此,每当有元素进栈时,temp_sum还表示上一轮的出栈元素对应集合的所有元素之和,leftsum_stack也压入当前状态下的temp_sum。

因此有:

temp_sum = 0;
while(遍历集合) {
  if(stack为空 or 当前元素>=stack栈顶元素) {
    //说明当前元素的左边界到stack栈顶元素就截止了
    //此时的temp_sum的值表示当前元素的左侧所有元素之和

    当前元素入栈stack;
    temp_sum入栈leftsum_stack;

    temp_sum = 0;
      //无论下一个遍历的是什么样的元素,
      //它的左边界都一定是现在这个栈顶元素,temp_sum要从0开始记起

    移动到下一个元素;
  }
  else { //此时当前元素 < stack栈顶元素

    temp_sum = stack栈顶元素 + leftsum_stack栈顶元素+temp_sum;
      //翻译一下就是:当前元素对应的集合元素之和 = 当前元素 + 左侧元素之和 + 右侧元素之和

    计算stack栈顶元素*temp_sum,并与之前循环得到的结果进行比较,取最大值
    stack出栈
    leftsum_stack出栈
  }
}

3. 得到代码

至此,所有推理都完成了,上面的例子的运行过程如下图。


然后就是激动人心的写代码时间了!

#include <stdio.h>
#include <stack>
using namespace std;

int main() {
    int N;
    scanf("%d", &N);
    int set[N + 1];
    for (int i = 0; i < N; i++)
        scanf("%d", &set[i]);
    set[N] = 0; //在集合末尾多添加一个元素,设置为0,在栈的操作中会用到
    
    stack<int> st;
    int i = 0;
    int temp_sum = 0; //当前子集的和
    stack<int> leftsum_stack;
    int max_result = 0;
    while (i <= N) {
        if (st.empty() || set[i] >= st.top()) { //如果栈为空或该元素大于栈顶元素
            st.push(set[i]); //让当前元素入栈
            leftsum_stack.push(temp_sum);
            temp_sum = 0;
            i++;
        }
        else { //当前元素小于栈顶元素
            temp_sum = st.top() + leftsum_stack.top() + temp_sum;
            if (temp_sum * st.top() > max_result)
                max_result = temp_sum * st.top();
            st.pop();
            leftsum_stack.pop();
        }
    }
    printf("%d", max_result);
    return 0;
}

全部AC。


最后,感谢无私分享自己解题方法的大佬们和愿意把这篇罗里吧嗦的文章阅读到最后的你,如果哪里有错误,欢迎指出。最后,努力的人运气通常都不会太差!

这篇文章写起来不易,如果它对你有帮助的话,麻烦给个赞吧!!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,588评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,456评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,146评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,387评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,481评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,510评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,522评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,296评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,745评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,039评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,202评论 1 343
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,901评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,538评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,165评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,415评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,081评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,085评论 2 352

推荐阅读更多精彩内容

  • 七.数组 1.给定一个整数数组 nums和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返...
    寒江_d764阅读 281评论 0 0
  • 目录 1 左神部分集锦 2 Leetcode前150题 3 牛客网剑指offer 4 JavaG 5 题目中的...
    小小千千阅读 984评论 0 0
  • 程序设计基础(上) 前言 C++的常见错误 语法错误: 混淆结构体类型与结构体变量的区别,为一个结构体类型赋值。不...
    修Nya阅读 1,257评论 0 0
  • 动态规划 111. 爬楼梯思路类似斐波那契数列注意考虑第 0 阶的特殊情况 272. 爬楼梯 II思路类似上题,只...
    6默默Welsh阅读 2,426评论 0 1
  • 夜莺2517阅读 127,717评论 1 9