Luogu 2839 [国家集训队]middle

孤街醉人 提交于 2020-01-02 12:08:44

感觉这题挺好的。

首先对于中位数最大有一个很经典的处理方法就是二分,每次二分一个数组中的下标$mid$,然后我们把$mid$代回到原来的数组中检查,如果一个数$a_{i} \geq mid$,那么就把$s_{i}$记为$1$,否则把$s_{i}$记为$-1$,然后对$s_{i}$跑一遍前缀和,观察是否有一个区间的和不小于$0$。

读清楚题意之后发现在这题中,如果要对一个长度为偶数(记为$n$)的序列求中位数,那么答案为排好序的数组中下标为$n / 2 + 1$的元素。(下标从$1$开始),不同的中位数$s_{i}$的表示方法可能有少许不同,要具体题目具体yy。

证明应当也很简单,$+1$代表一个比当前的$mid$大的数,而$-1$表示一个比当前的$a_{mid}$小的数,那么当$a_{mid}$是中位数的条件在一个区间成立的时候,这个区间的$s_{i}$和应当恰好等于$0$。而当得到了一个区间和大于$0$的区间的时候,我们可以通过扔掉几个$1$使它变成$0$,相当于之前的条件成立。

但是这样还是太慢了。

对于本题来说,每次二分得到了一个$mid$,我们检验的时候得到的答案就一定是(区间是$a, b, c, d$)$sum(b + 1, c - 1) + lmax(c, d) + rmax(a, b)$。

其中$lmax$代表一定要选左端点的最大子段和,而$rmax$代表一定要选右端点的最大子段和。

显然可以用线段树来维护。

一个元素开一颗线段树是不可能的……

我们发现在排好序中的数组里面两个下标相差$1$的元素的值其实只有$1$个不同,这就是较小的元素在原数组中的下标,那么大多数结点的值可以继承过来。

于是可以可持久化了,变成了一个主席树。

时间复杂度$O(nlog^{2}n)$。

Code:

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

const int N = 2e4 + 5;
const int inf = 1 << 30;

int n, qn, a[N];

struct Innum {
    int val, id;
} b[N];

bool cmp(const Innum &x, const Innum &y) {
    if(x.val != y.val) return x.val < y.val;
    else return x.id < y.id;
}   

/*bool cmp(const Innum &x, const Innum &y) {
    return x.val < y.val;
}   */

inline void read(int &X) {
    X = 0; char ch = 0; int op = 1;
    for(; ch > '9' || ch < '0'; ch = getchar())
        if(ch == '-') op = -1;
    for(; ch >= '0' && ch <= '9'; ch = getchar())
        X = (X << 3) + (X << 1) + ch - 48;
    X *= op;
}

inline int max(int x, int y) {
    return x > y ? x : y;
}

inline void chkMax(int &x, int y) {
    if(y > x) x = y;
}

namespace PSegT {
    struct Node {
        int lc, rc, sum, lmax, rmax;
        
        inline void init() {
            lc = rc = sum = 0;
            lmax = rmax = -inf;
        }
        
    } s[N * 20];
    
    int root[N], nodeCnt = 0;
    
    #define lc(p) s[p].lc
    #define rc(p) s[p].rc
    #define sum(p) s[p].sum
    #define lmax(p) s[p].lmax
    #define rmax(p) s[p].rmax
    #define mid ((l + r) >> 1)
    
    inline void up(int p) {
        if(!p) return;
        sum(p) = sum(lc(p)) + sum(rc(p));
//        lmax(p) = rmax(p) = -inf;
        lmax(p) = max(lmax(lc(p)), lmax(rc(p)) + sum(lc(p)));
        rmax(p) = max(rmax(rc(p)), rmax(lc(p)) + sum(rc(p)));
    }

    void build(int &p, int l, int r) {
        p = ++nodeCnt;
        if(l == r) {
            sum(p) = lmax(p) = rmax(p) = 1;
            return;
        }
        
        build(lc(p), l, mid);
        build(rc(p), mid + 1, r);
        up(p);
    }
    
    void modify(int &p, int l, int r, int x, int v, int pre) {
        s[p = ++nodeCnt] = s[pre];
        if(l == r) {
            sum(p) = lmax(p) = rmax(p) = v;
            return;
        }
        
        if(x <= mid) modify(lc(p), l, mid, x, v, lc(pre));
        else modify(rc(p), mid + 1, r, x, v, rc(pre));
        up(p);
    }
    
    Node query(int p, int l, int r, int x, int y) {
        if(x <= l && y >= r) return s[p];
        
        Node res, ln, rn; 
        res.init(), ln.init(), rn.init();
        
        if(x <= mid) ln = query(lc(p), l, mid, x, y);
        if(y > mid) rn = query(rc(p), mid + 1, r, x, y);
        
        res.sum = ln.sum + rn.sum;
        res.lmax = max(ln.lmax, ln.sum + rn.lmax);
        res.rmax = max(rn.rmax, rn.sum + ln.rmax);
        
        return res;
    }
    
    #undef lc 
    #undef rc 
    #undef sum 
    #undef lmax 
    #undef rmax 
    #undef mid 
    
} using namespace PSegT;

inline bool chk(int mid, int l1, int r1, int l2, int r2) {
    Node now; int res = 0;
    
    if(r1 + 1 <= l2 - 1) {
        now.init();
        now = query(root[mid], 1, n, r1 + 1, l2 - 1);
        res += now.sum; 
    }
    
    now.init();
    now = query(root[mid], 1, n, l1, r1);
    res += now.rmax;
    
    now.init();
    now = query(root[mid], 1, n, l2, r2);
    res += now.lmax;
    
    return res >= 0;
}

inline int solve(int l1, int r1, int l2, int r2) {
    int ln = 1, rn = n, mid, res;
    for(; ln <= rn; ) {
        mid = (ln + rn) / 2;
        if(chk(mid, l1, r1, l2, r2)) res = mid, ln = mid + 1;
        else rn = mid - 1;
    }
    return b[res].val;
}

int main() {
    read(n);
    for(int i = 1; i <= n; i++) {
        read(a[i]);
        b[i].val = a[i], b[i].id = i;
    }
    
    s[0].lmax = s[0].rmax = -inf;
    build(root[1], 1, n);
    sort(b + 1, b + 1 + n, cmp);
    for(int i = 2; i <= n; i++)
        modify(root[i], 1, n, b[i - 1].id, -1, root[i - 1]);
    
    read(qn);
    for(int ans = 0, q[4]; qn--; ) {
        for(int i = 0; i < 4; i++) {
            read(q[i]);
            q[i] = (q[i] + ans) % n + 1;
        }
        sort(q, q + 4);
        ans = solve(q[0], q[1], q[2], q[3]);
        printf("%d\n", ans);
    }    
    
    return 0;
}
View Code

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!