线段树优化贪心策略。
先考虑
猜结论。将
再做一遍后缀和,执行同样的操作。
最后的答案就是区间长度
事实上,交一发,发现有 40pts,证明贪心是对的。
(但是我真的不会证贪心策略,我只能感性理解,如果有神仙知道请发到评论区 /kk)
那么考虑答案的变化过程。
比方说,当前前缀和数组
继续碰到
那么我们发现,统计做前缀和的过程,其实依次就是找到第一个
那么第一轮操作(统计前缀和)的操作次数,也就是
由于第一轮操作会对第二轮操作(统计后缀和)产生影响,考虑影响。
对于现在的后缀数组
想到这个不好维护,转化为差,即
根据上侧的贪心策略,在
那么
则最终答案为
考虑怎么维护这个东西。
想想,答案其实是对于每一个段
枚举
问题就转化为,求互不相交的一个前缀和一个后缀的和的最小值。
继续转化,其实一个前缀和一个后缀的和的最小值,就等于全局和减去夹在中间的区间和。
问题等于
那么就转化为最大子段和问题。这个东西显然可以推广到区间上。
由于题目问的是最多剩下多少个数,所以要用区间长度减掉维护的答案。
#include <bits/stdc++.h>
#define ls ((rt) << 1)
#define rs ((rt) << 1 | 1)
using namespace std;
const int N = 5e5 + 10; int n, q;
struct Node { int al, sum, pre, suf; } t[N << 2];
inline Node pushup(Node f, Node s) {
Node res; res.sum = f.sum + s.sum;
res.pre = max(f.pre, f.sum + s.pre), res.suf = max(s.suf, s.sum + f.suf);
res.al = max(max(f.al, s.al), f.suf + s.pre); return res;
}
inline void build(int rt, int l, int r) {
if (l == r) {
char op; cin >> op;
t[rt].al = t[rt].sum = t[rt].pre = t[rt].suf = (op == '0'? -1 : 1);
if (t[rt].al < 0) t[rt].al = 0; return ;
}
int mid = (l + r) >> 1;
build(ls, l, mid), build(rs, mid + 1, r);
t[rt] = pushup(t[ls], t[rs]); return ;
}
inline Node query(int rt, int l, int r, int L, int R) {
if (L <= l && r <= R) return t[rt];
int mid = (l + r) >> 1; Node f, s; int tag = 0;
if (L <= mid) f = query(ls, l, mid, L, R), ++tag;
if (R > mid) s = query(rs, mid + 1, r, L, R), tag += 2;
if (tag == 1) return f; if (tag == 2) return s;
return pushup(f, s);
}
inline void solve() {
int l, r; cin >> l >> r;
Node res = query(1, 1, n, l, r); int len = res.al - res.sum;
if (len >= r - l + 1) cout << -1 << endl;
else cout << r - l + 1 - len << endl; return ;
}
int main() {
ios_base::sync_with_stdio(false); cin.tie(0), cout.tie(0);
cin >> n >> q; build(1, 1, n);
while (q--) solve(); return 0;
}