[COI2007] Patrik 音乐会的等待

[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;
}
上一篇:316. 去除重复字母


下一篇:学习心得和体会