省队集训 Day5 选举

【题目大意】

小奇和魔法猪要竞选膜钟国的总统。

有 $n$ 个选民，编号为$1...n$，他们中有的人支持小奇，有的人支持魔法猪，还有的人保持中立。

现在你需要把选民分成若干个区间，每个区间的长度在$[l,r]$中。如果一个区间中支持小奇的人比支持魔法猪的人多，那么小奇得一票，一个区间中支持魔法猪的人比支持小奇的人多，那么魔法猪得一票。

小奇想要赢得选举，于是它请你合理地分区间，使得它获得的票数减去魔法猪获得的票数最大。如果你帮它解决了这个问题，它会送你这道题的题解。

$n leq 10^6$

【题解】

考虑暴力dp，复杂度为$O(n^2)$。

然后考虑优化暴力，每次相当于区间往右移动一格，相当于加入1个数，减去1个数，这个可以用线段树来维护

如果普通写法需要开单调队列，我们可以把下标映射到位置，这样就能执行减去1个数的操作了。

复杂度$O(nlogn)$

# include <vector>
# include <stdio.h>
# include <string.h>
# include <iostream>
# include <algorithm>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;

const int M = 5e5 + 10, N = 1e6 + 10, inf = 1e9, SN = 1024 * 1024 + 5;

inline int getint() {
    int x = 0, f = 1; char ch = getchar();
    while(!isdigit(ch)) {
        if(ch == '-') f = 0;
        ch = getchar();
    }
    while(isdigit(ch)) x = (x<<3) + (x<<1) + ch - '0', ch = getchar();
    return f ? x : -x;
}

int n, L, R, a[N], s[N], f[N], pos[N], idx;
vector<int> v[N + N]; int l[N], r[N];

struct SMT {
    int w[N << 2];
    # define ls (x<<1)
    # define rs (x<<1|1)
    inline void up(int x) {
        w[x] = max(w[ls], w[rs]);
    }
    inline void build(int x, int l, int r) {
        w[x] = -inf;
        if(l == r) return ;
        int mid = l+r>>1;
        build(ls, l, mid); build(rs, mid+1, r);
    }
    inline void edt(int x, int l, int r, int pos, int del) {
        if(l == r) {
            w[x] = del;
            return ;
        }
        int mid = l+r>>1;
        if(pos <= mid) edt(ls, l, mid, pos, del);
        else edt(rs, mid+1, r, pos, del);
        up(x);
    }
    inline int query(int x, int l, int r, int L, int R) {
        if(L > R) return -inf;
        if(L <= l && r <= R) return w[x];
        int mid = l+r>>1;
        if(R <= mid) return query(ls, l, mid, L, R);
        else if(L > mid) return query(rs, mid+1, r, L, R);
        else return max(query(ls, l, mid, L, mid), query(rs, mid+1, r, mid+1, R));
    }
    # undef ls
    # undef rs
}T;

inline void ins(int i) {
//    cout << "add " << pos[i] << endl;
    T.edt(1, 0, n, pos[i], f[i]);
}

inline void del(int i) {
    T.edt(1, 0, n, pos[i], -inf);
}

int main() {
    freopen("election.in", "r", stdin);
    freopen("election.out", "w", stdout);
    n = getint(), L = getint(), R = getint();
    for (int i=1; i<=n; ++i) a[i] = getint();
    for (int i=1; i<=n; ++i) s[i] = s[i-1] + a[i];
//    for (int i=0; i<=n; ++i) cout << s[i] << ' '; cout << endl;
    
    for (int i=0; i<=n; ++i) v[s[i] + n].push_back(i);
    for (int i=0; i<=n+n; ++i) {
        int lst = idx;
        for (int j=0; j<v[i].size(); ++j) 
            pos[v[i][j]] = idx++;
        for (int j=0; j<v[i].size(); ++j)
            l[v[i][j]] = lst, r[v[i][j]] = idx - 1;
    }
//    for (int i=0; i<=n; ++i) cout << pos[i] << ' ' << l[i] << ' ' << r[i] << endl; cout << endl;
    
    f[0] = 0;
    T.build(1, 0, n);
    for (int i=1; i<=n; ++i) {
        if(i - R - 1 >= 0) del(i - R - 1);
        if(i - L >= 0) ins(i - L);
        int ta = T.query(1, 0, n, 0, l[i]-1), tb = T.query(1, 0, n, r[i]+1, n), tc = T.query(1, 0, n, l[i], r[i]);
        if(ta != -inf) ta ++;
        if(tb != -inf) tb --;
//        cout << i << "???" << l[i]-1 <<endl;
//        cout << i << ' ' << ta << ' ' << tb << ' ' << tc << endl;
        f[i] = max(ta, max(tb, tc));
    }
        
//    }
    if(f[n] == -inf) cout << "Impossible";
    else cout << f[n];
    return 0;
}