糟糕的一天【栈】
题目大意:
农民约翰的某N头奶牛正在过乱头发节!由于每头牛都意识到自己凌乱不堪的发型,约翰希望统计出能够看到其他牛的头发的牛的数量。
每一头牛i有一个高度所有N头牛面向东方排成一排,牛N在最前面,而 牛1在最后面.第i头牛可以看到她前面的那些牛的头,只要那些牛的高度严格小于她的高度,而且 中间没有比hi高或相等的奶牛阻隔.
让N表示第i头牛可以看到发型的牛的数量;请输出Ci的总和。
思路:
很明显,这道题是一个单调栈。
将奶牛一个一个进栈,同时保证栈里面的数字单调递减(后面必须看得见前面),每次进栈之后求出最后面一头牛能看见前面的牛的数量,输出答案即可。
时间复杂度:O(n)
代码:
#include <cstdio> #include <iostream> #define fre(x) freopen(#x".in","r",stdin),freopen(#x".out","w",stdout); using namespace std; long long s[100011],n,a[100011],sum,tail; int main() { fre(badhair); cin>>n; tail=0; a[0]=1e17; for (long long i=1;i<=n;i++) cin>>a[i]; for (long long i=1;i<=n;i++) { while (a[i]>=s[tail]&&tail) tail--; //模拟栈(可以用queue) tail++; s[tail]=a[i]; //进栈 sum=sum+tail-1; //求出答案 } cout<<sum<<"\n"; return 0; }