P2866 [USACO06NOV]Bad Hair Day S
奶牛的发型
题目传送门
一、理解和感悟
1、号码和值都是非常重要的,不要只记录值,用一个数组\(a[i]\)来保存值,号码就是\(i\)。
2、每头牛都是在找到了右侧比自己高的牛时,记录右侧第一个比自己高的牛。
3、最终统计时有两种情况,一种是找到了右侧比自己高的牛,另一种是没有找到右侧比自己高的牛。
4、找到了右侧比自己高的牛,那么\(ans+=j-i-1\),简单植树问题两棵树之间夹着几棵树。
5、没有找到右侧比自己高的牛,那么就是它向右可以一直看到头,即\(n-i\)个。
二、完整代码
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 80010;
stack<int> stk;
LL a[N], res[N];
int n;
LL ans;
/**
输入样例
6
10 3 7 4 12 2
找出右侧第一个比自己大的数字
原值:10 3 7 4 12 2
原号:1 2 3 4 5 6
对值:12 7 12 12 -1 -1
对号:5 3 5 5 -1 -1
*/
int main() {
cin >> n;
for (int i = 1; i <= n; i++)cin >> a[i];
for (int i = 1; i <= n; i++) {
while (!stk.empty() && a[stk.top()] <= a[i]) {
res[stk.top()] = i;
stk.pop();
}
stk.push(i);
}
for (int i = 1; i <= n; i++)
if (res[i]) ans += res[i] - i - 1;//两个号之间夹的,需要j-i-1,比如4和2之间,其实就是一个3,就是4-2-1=1
else ans += n - i; //如果右侧没有比自己大的,那么就是一直到最右端,它都能看到。
//输出结果
cout << ans << endl;
return 0;
}