LeetCode 1944. 队列中可以看到的人数（单调栈）

输入：heights = [10,6,8,5,11,9]
输出：[3,1,2,1,1,0]
解释：
第 0 个人能看到编号为 1 ，2 和 4 的人。
第 1 个人能看到编号为 2 的人。
第 2 个人能看到编号为 3 和 4 的人。
第 3 个人能看到编号为 4 的人。
第 4 个人能看到编号为 5 的人。
第 5 个人谁也看不到因为他右边没人。

示例 2：
输入：heights = [5,1,2,3,10]
输出：[4,1,1,1,0]
 
提示：
n == heights.length
1 <= n <= 10^5
1 <= heights[i] <= 10^5
heights 中所有数 互不相同 。

class Solution {
public:
    vector<int> canSeePersonsCount(vector<int>& heights) {
        int n = heights.size();
        vector<int> ans(n, 0);
        stack<int> s;
        for(int i = n-1; i >= 0; i--)
        {
            bool flag = false;//是否见过比自己矮的
            while(!s.empty() && heights[i] > heights[s.top()])
            {
                s.pop();//后续的人肯定看不见
                flag = true;
                ans[i]++;//见过一个比自己矮的，+1，可以看见
            }
            if(!flag && i != n-1)
                ans[i] = 1;//除了最右边的人，至少可以看见1个人
            else if(flag && !s.empty())
                ans[i]++;//如果见过比自己矮的人，且栈内还有人（比自己高，能看见）
            s.push(i);
        }
        return ans;
    }
};