采用单调栈,从后向前遍历,栈顶元素与当前元素比较
1)当前元素 > 栈顶元素 ,说明当前元素会被打中,将当前元素位置加到结果res中;取出栈顶元素;继续取新的栈顶元素比较;将当前元素入栈
2)当前元素 < 栈顶元素 ,说明当前元素不会被打中,直接将当前元素入栈。
class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param n int整型 n个人 * @param a int整型vector ai代表第i个人的高度 * @return long长整型 */ long long solve(int n, vector<int>& a) { // write code here stack<int> st; long long res = 0; for (int i = n - 1; i >= 0; i--) { if (st.empty()) { st.push(a[i]); } else { while (!st.empty() && a[i] > st.top()) { res += (i + 1); st.pop(); } st.push(a[i]); } } return res; } };