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

Posted 2020-09-29 third2333

洛谷P1823 音乐会的等待
单调栈
维护一个上升的单调栈
用以记录有当前这个数向后能看到几个数
但是每次加入一个数 时 他能看到的 是
单调栈中所有比他小的 和跟他一样的数
比他小的下次就没有用了，所以直接退栈
但是 相同的数到后面还是可能会有贡献的，所以贡献算完以后
又要进栈
最后如果栈中还有元素，那么栈顶一定能看到自身，所以+1

 

 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 }