[H单调栈] lc5196. 队列中可以看到的人数(单调栈+双周赛57_4)

Posted 2021-08-27 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：5196. 队列中可以看到的人数

2. 题目解析

单调栈裸题。

和朴素的找他右边的第一个比它高的数不太一样，因为对于样例一而言，编号 3 是看不到的。

但不难发现，从后至前维护一个单调递增栈，对于当前元素而言，比它小的栈中元素就是它能看到的元素，也就是维护单调递增栈时的出栈元素个数。故维护过程中记录下出栈元素个数即可。

注意，当栈不为空时，说明栈中存在一个比当前元素还高的元素，它也是能看到的，所以需要将出栈个数 +1，这才是当前元素所能看到的所有元素个数。

最后记得反转一下整个答案数组，因为是倒着做的。

---

本次双周赛异常简单，都是些基本算法和裸题…手速场了属于是。

---

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

---

代码：

我的代码。

思路差不多，关于两边端点的问题，可以使用标记数组处理，即对差分过的端点位置打个标记，最后直接求就行了。大同小异吧。

class Solution {
public:
    vector<int> canSeePersonsCount(vector<int>& h) {
        int n = h.size();
        stack<int> stk;
        vector<int> res;
        for (int i = n - 1; ~i; i -- ) {
            int cnt = 0;
            while (stk.size() && h[stk.top()] <= h[i]) stk.pop(), cnt ++ ;
            if (stk.size()) res.push_back(cnt + 1);
            else res.push_back(cnt);
            stk.push(i);
        }
        reverse(res.begin(), res.end());
        
        return res;
    }
};

---

liouzhou_101 大佬的代码。

思路差不多，但是实现更加简洁。

class Solution {
public:
    vector<int> canSeePersonsCount(vector<int>& h) {
        int n = h.size();
        vector<int> v;
        vector<int> res(n);
        for (int i = n - 1; i >= 0; --i)
        {
            while (!v.empty() && v.back() < h[i])
            {
                v.pop_back();
                res[i] += 1;
            }
            if (!v.empty()) res[i] += 1;
            v.push_back(h[i]);
        }
        return res;
    }
};