采用单调栈,从后向前遍历,栈顶元素与当前元素比较
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;
}
};

京公网安备 11010502036488号