hfut 1287

http://acm.hfut.edu.cn/OnlineJudge/

中文题。

区间线段树，需要剪枝。n的大小有问题。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>

using namespace std;
#define ls rt<<1
#define rs rt<<1|1
#define lson l, m, rt<<1
#define rson m + 1, r, rt<<1|1
const int inf = 0x3f3f3f3f;
const int maxn = 2e4 + 5;

struct SegTree{
    int maxv, minv;
    int lx, rx, sumx;//该区间左边是否被淹，右边，总的小岛数
}seg[maxn << 2];
struct Node{
    int h, id;
}q[maxn];
int a[maxn];

void build(int l, int r, int rt){
    seg[rt].lx = seg[rt].rx = seg[rt].sumx = 1;
    if (l == r){
        seg[rt].maxv = seg[rt].minv = a[l];
        return ;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    seg[rt].maxv = max(seg[ls].maxv, seg[rs].maxv);
    seg[rt].minv = min(seg[ls].minv, seg[rs].minv);
}
void pushUp(int rt){
    seg[rt].sumx = seg[ls].sumx + seg[rs].sumx - (seg[ls].rx & seg[rs].lx);
    seg[rt].lx = seg[ls].lx;
    seg[rt].rx = seg[rs].rx;
    seg[rt].minv = min(seg[ls].minv, seg[rs].minv);//
}
void update(int h, int l, int r, int rt){
    if (h < seg[rt].minv) return;
    if (h >= seg[rt].maxv){
        seg[rt].minv = inf;//这句表明接下来的h对这个区间都不会再有影响，相当于减掉了这个区间否则tle
        seg[rt].rx = seg[rt].lx = seg[rt].sumx = 0;
        return ;
    }
    int m = (l + r) >> 1;
    update(h, lson);
    update(h, rson);
    pushUp(rt);
}
bool cmp(Node a, Node b){
    return a.h < b.h;
}
int ans[maxn];
int main(){
    int n, m;
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i){
        scanf("%d", &a[i]);
    }
    build(1, n, 1);
    scanf("%d", &m);
    for (int i = 0; i < m; ++i){
        scanf("%d", &q[i].h);
        q[i].id = i;
    }
    sort(q, q + m, cmp);
    for (int i = 0; i < m; ++i){
        update(q[i].h, 1, n, 1);
        ans[q[i].id] = seg[1].sumx;
    }
    for (int i = 0; i < m; ++i){
        printf("%d
", ans[i]);
    }
    return 0;
}