「CF484E」Sign on Fence「整体二分」「线段树」

题意

给定一个长度为\(n\)的正整数序列,第\(i\)个数为\(h_i\),\(m\)个询问,每次询问\((l, r, w)\),为\([l, r]\)所有长度为\(w\)的子区间最小值的最大值。(类似于一类特殊的直方图最大子矩形问题)

\(1 \leq n, m \leq 10^5\)

题解

我们考虑二分答案,这样\(n\)个数变成\(01\),若\(h_i\geq mid\)则为\(0\),否则为\(1\)

每次就相当于查询存不存在长度为\(w\)的连续\(1\)。用线段树维护。

这有个问题,\([l, r]\)分成\([l, mid - 1]\)和\([mid, r]\)的时候,左区间统计不到右区间的贡献。那我们就递归左区间之前不清空线段树,等到递归右区间的时候再清空。

时间复杂度两个log

#include <algorithm>
#include <cstdio>
using namespace std;

const int N = 2e5 + 10;

struct opt { int l, r, k, id; } q[N], qL[N], qR[N];
int n, m, h[N], ans[N];
struct node { int res, l, r, len; } t[N << 2];

node operator + (const node &a, const node &b) {
    node ans; ans.len = a.len + b.len;
    ans.l = a.l == a.len ? a.l + b.l : a.l;
    ans.r = b.r == b.len ? b.r + a.r : b.r;
    ans.res = max(max(a.res, b.res), a.r + b.l);
    return ans;
}

void build(int u, int l, int r) {
    if(l == r) { t[u] = (node) {0, 0, 0, 1}; return ; }
    int mid = (l + r) >> 1;
    build(u << 1, l, mid);
    build(u << 1 | 1, mid + 1, r);
    t[u] = t[u << 1] + t[u << 1 | 1];
}

void ins(int u, int l, int r, int x, int y) {
    if(l == r) { t[u] = (node) {y, y, y, 1}; return ; }
    int mid = (l + r) >> 1;
    if(x <= mid) ins(u << 1, l, mid, x, y);
    else ins(u << 1 | 1, mid + 1, r, x, y);
    t[u] = t[u << 1] + t[u << 1 | 1];
}

node qry(int u, int l, int r, int ql, int qr) {
    if(l == ql && r == qr) return t[u];
    int mid = (l + r) >> 1;
    if(qr <= mid) return qry(u << 1, l, mid, ql, qr);
    if(ql > mid) return qry(u << 1 | 1, mid + 1, r, ql, qr);
    return qry(u << 1, l, mid, ql, mid) + qry(u << 1 | 1, mid + 1, r, mid + 1, qr);
}

void solve(int ql, int qr, int l, int r) {
    if(ql > qr || l > r) return ;
    if(l == r) {
        for(int i = ql; i <= qr; i ++) ans[q[i].id] = l;
        return ;
    }
//  printf("[%d, %d] & [%d, %d]\n", ql, qr, l, r);
    int mid = (l + r + 1) >> 1, nl = 0, nr = 0;
    for(int i = ql; i <= qr; i ++) {
        if(!q[i].id) {
            if(q[i].k >= mid) ins(1, 1, n, q[i].l, 1), qR[nr ++] = q[i];
            else qL[nl ++] = q[i];
        } else {
            int res = qry(1, 1, n, q[i].l, q[i].r).res;
            if(res >= q[i].k) qR[nr ++] = q[i];
            else qL[nl ++] = q[i];
        }
    }
    for(int i = 0; i < nl; i ++) q[ql + i] = qL[i];
    for(int i = 0; i < nr; i ++) q[ql + nl + i] = qR[i];
    solve(ql, ql + nl - 1, l, mid - 1);
    for(int i = ql + nl; i <= qr; i ++) if(!q[i].id && q[i].k >= mid) ins(1, 1, n, q[i].l, 0);
    solve(ql + nl, qr, mid, r);
}
int main() {
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++) scanf("%d", h + i), q[i] = (opt) {i, 0, h[i], 0};
    int l = *min_element(h + 1, h + n + 1);
    int r = *max_element(h + 1, h + n + 1);
    scanf("%d", &m);
    for(int i = n + 1; i <= n + m; i ++) {
        scanf("%d%d%d", &q[i].l, &q[i].r, &q[i].k); q[i].id = i - n;
    }
    build(1, 1, n); solve(1, n + m, l, r);
    for(int i = 1; i <= m; i ++) printf("%d\n", ans[i]);
    return 0;
}
上一篇:Java学习-035-JavaWeb_004 -- JSP include 指令


下一篇:CF786B Legacy