问题 A: 乱发节(bzoj1660)
时间限制: 1 Sec 内存限制: 128 MB
提交: 142 解决: 52
[提交][状态]
题目描述
问题描述 农民约翰的N(1<=N<=80000)头奶牛正在过乱头发节!由于每头奶牛都意识到自己凌乱不堪的发型,约翰希望统计出能够看到其它牛的头发的牛的数量。 每一头牛i有一个高度hi( 1 <=hi <=10^9)。所有N头牛面向东方排成一排,牛N在最前面,而牛1在最后面,第i头牛可以看到它前面的那些牛的头,只要那些牛的高度严格小于它的高度,而且中间没有比hi高的或相等的奶牛阻隔。 让c[i]表示第i头牛可以看到发型的牛的数量,请输出c[1]+c[2]+…+c[n]。
输入
第1行:牛的数量 N。 第2到N+1:第 i+1 是一个整数,表示第i头牛的高度。
输出
一个整数表示c[1] 至 c[N]的和。
样例输入
6
10
3
7
4
12
2
输入解释:六头牛排成一排,高度依次是 10, 3, 7, 4, 12, 2。
样例输出
5
提示
输出解释:3+0+1+0+1=5
林肯是大头:http://www.accoders.com/problem.php?cid=1965&pid=0
思路:
1.从1到N枚举,维护一个单调递减的栈,在出栈时(被挡住了)即可把i-s[t]-1加到答案中
2.从N到1枚举,同样维护一个单调递减的栈,此时入站时弹出几头牛答案就加几,直到遇到第一个比它高的(被挡住了)
注意:
1.一次WA了是因为未考虑到取等
2.答案long long存
3.单调栈问题易错点:枚举完N后栈内可能还有元素
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=80000+10;
int a[maxn];
int s[maxn];
int t=0;
long long ans=0;
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
a[0]=0x7fffffff;//单调栈边界
for(int i=1;i<=n;i++)
{
while(a[s[t]]<=a[i])//取等!
{
ans+=1ll*(i-1-s[t]);
t--;
}
s[++t]=i;
// printf("%d %d %d %lld\n",i,t,s[t],ans);
}
while(t)//注意!栈内还有可能有元素
{
ans+=(n-s[t]);
t--;
}
printf("%lld",ans);
return 0;
}