原题链接
Farmer John的奶牛在风中凌乱了它们的发型……
每只奶牛都有一个身高hi(1 ≤ hi ≤ 1,000,000,000),现在在这里有一排全部面向右方的奶牛,一共有N只(1 ≤ N ≤ 80,000)。对于奶牛i来说,如果奶牛i+1,i+2,……,N这些奶牛的身高严格小于奶牛i,则奶牛i可以看到它们凌乱的发型。
比如下面这个例子:
=
= =
= * - * = * 奶牛面向这边–>
= * = * = *
= - = = = *
= = = = = =
1 2 3 4 5 6
令ci表示第i只奶牛能够看到的发型数量,请计算c1 + c2 + c3 + … + cN的值
对于上面这个例子,答案为3 + 0 + 1 + 0 + 1 + 0=5。
输入
第一行:奶牛数量N
第二到N+1行:第i+1行输入奶牛i的身高
输出
第一行:一个整数即c1到cN的和
样例输入
6
10
3
7
4
12
2
样例输出
5
题目描述
一群身高不等的奶牛排成一排,向右看,每个奶牛只能看到身高小于自己的奶牛发型,问这些奶牛能够看到的发型总和是多少
思路:利用单调栈(栈中元素从栈顶往下越来越大)的思想,可以计算出,每只奶牛能被他前面的多少只奶牛看到自己的发型,就反向得到了奶牛看到的发型总和
借用别人的解释:首先弹出元素是因为它右边相邻牛比它高(看不到它的头发并且挡住了该牛的视线),“挡住”这个关键字很重要,这就是说该牛已经看不到后面的其他牛的头发啦,而思路一是按顺序比较身高,统计每头牛前面能看到它头发的牛数,既然该牛望不到后面,那么把它出栈对整个结果没有影响。
例如:10 3 7 4 12 2
①Height_List:10 入栈首身高
②3<10,不弹出,num=0+1(当前栈中元素数),3入栈后 Height_List:10 3
③7 > 3(3 弹出) 7<10(10保留),栈中剩 10 ,num=1+1,7入栈后 Height_List:10 7
④4 < 7,没有弹出,栈中10 7都能看到4,num=2+2 ,4入栈后 Height_List:10 7 4
⑤12大于栈中全部元素,表示栈中所有元素看不到12,全部出栈,num不变,12入栈后 Height_List:12
⑥2 < 12,没有弹出,12能看到2,num=4+1
最后结果num=5
代码如下
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <stack>
using namespace std;
stack<long long> st;
int a[80005];
int main()
{
int i,n;
long long sum = 0;
scanf("%d",&n);
for(i = 0;i < n;i++)
scanf("%d",&a[i]);
for(i = 0;i < n;i++)
{
while(!st.empty() && st.top() <= a[i]) //如果栈顶元素小于即将入栈元素,那么就不断出栈
st.pop();
sum += st.size();//st.size()表示将要入栈的元素,它后面能看到他发型的奶牛个数
st.push(a[i]);
}
printf("%lld\n",sum);
return 0;
}