[COI2007] Patrik 音乐会的等待
这题显然是维护一个递减的单调栈,但是要注意身高相等的情况的处理。
#include <stdio.h>
const int MAXN = 500010;
int n, tp, stk[MAXN];
long long ans = 0;
int main() {
scanf("%d", &n);
int tmp, same = 0;
for (int i = 1; i <= n; ++i) {
scanf("%d", &tmp);
while (tp && tmp > stk[tp]) {
ans++;
tp--;
}
if (tp) {
if (tmp == stk[tp]) {
same++;
ans += same;
}
else same = 0;
if (same < tp) ans++;
}
else same = 0;
stk[++tp] = tmp;
}
printf("%lld", ans);
return 0;
}
这样做实际上是有问题的
hack数据
7
2 2 2 1 2 2 2
当两群身高相等的人被一个较矮的隔开时,会重新计数,这样左边那一群人就被忽略了。应当在记录身高的同时记录身高相同的人数。
#include <stdio.h>
const int MAXN = 500010;
int n, tp = 0;
long long ans = 0;
struct node{
int val, num;
node() {}
node(int _val, int _num) : val(_val), num(_num) {}
};
node stk[MAXN];
int main() {
freopen("P1823_6.in", "r", stdin);
scanf("%d", &n);
int tmp, same = 0;
for (int i = 1; i <= n; ++i) {
scanf("%d", &tmp);
while (tp && tmp > stk[tp].val) {
ans += stk[tp].num;
tp--;
}
if (tp && tmp == stk[tp].val) {
ans += stk[tp].num;
stk[tp].num++;
} else {
stk[++tp] = node(tmp, 1);
}
if (tp > 1) ans++;
}
printf("%lld", ans);
return 0;
}