洛谷P1823 音乐会的等待 单调栈

 1 #include <bits/stdc++.h> 
 2 #define For(i,j,k) for(int i=j;i<=k;i++) 
 3 #define LL long long
 4 using namespace std ; 
 5 
 6 const int N = 500011 ; 
 7 int n,x,top,t ; 
 8 int st[N] ;
 9 LL sum ;  
10 
11 inline int read() 
12 {
13     int x = 0 , f = 1 ; 
14     char ch = getchar() ; 
15     while(ch<‘0‘||ch>‘9‘) { if(ch==‘-‘) f = -1 ; ch = getchar(); } 
16     while(ch>=‘0‘&&ch<=‘9‘) { x = x * 10+ch-48 ; ch = getchar(); } 
17     return x * f ; 
18 }
19 
20 int main() 
21 {
22     n = read() ; 
23     For(i,1,n) {
24         x = read() ; 
25         t = 1 ; // 记录有几个数与 x 相同 包括自身 
26         while(top>0&&st[ top ] <= x ) {
27             if(st[top] == x ) t++ ;  
28             top-- ; sum++ ; 
29         } 
30         if( top>0 ) sum++ ; 
31         For(j,1,t) st[++top] = x ; 
32     }
33     printf("%lld",sum) ; 
34     return 0 ; 
35 }