解题思路
将每头牛加入单调栈,记录每次pop
操作后栈的长度,也就是每头牛能被多少牛看到,累加之后与每头牛能看到的牛的数量之和一致。
题目里\(1<=n<=80,000\),如果牛的身高序列单调递减,结果最大为\(79999+79998+...+1=79999*80000/2\approx 3.2*10^9\),超出了int
的范围,需要用long long
。
AC代码
#include <iostream>
#include <stack>
int main() {
int n;
std::cin >> n;
std::stack<int> stack;
int cow;
std::cin >> cow;
stack.push(cow);
long long res = 0;
for (int i = 1; i < n; i++) {
std::cin >> cow;
while (!stack.empty() && stack.top() <= cow) {
stack.pop();
}
res += stack.size();
stack.push(cow);
}
std::cout << res << std::endl;
return 0;
}