CSP-S 例题7-1 单调栈举例-看发型

大家排队，都面向右边。大家只能向下看.从左边数第一位开始，如果第2位同学比他低，他能看到该同学的头型，如果高就看不到了，后面的也看不到。输入n个人，再输入n个人的身高。求所有人能看到其他人的头型总数有多少(1<n<10^6)
输入

10 3 7 4 12 2

输出

5

#include<bits/stdc++.h>
using namespace std;
int main(){
	long long n,ans=0;
	stack<long long>s;
	cin>>n;
	while(n--){
		long long p;
		cin>>p;
		while(!s.empty()&&s.top()<=p){
			s.pop();
		}
		ans+=s.size();
		s.push(p);
	}
	cout<<ans;
	return 0;
}