2023牛客暑期多校训练营2 H题 0 and 1 in BIT

链接

题意：定义两种作用于二进制数$x$的事件，$A$事件：$x$变为$\sim x$（按位取反），$B$事件：$x$变为$x+1$。给一个长为$n(1\le n\le 2\times 10^5)$的事件数组，用只包含$AB$的字符串表示。接下来有$q(1\le q\le 2\times 10^5)$次询问，每次询问给出一个区间$[l,r](1\le l\le r\le n)$和二进制数$x(1\le len(x)\le 50)$，输出$x$经过$[l,r]$事件后变成的二进制数，强制在线$[l,r]$是用$last\_ans$加密的。

题解：

$len(x)\le 50$，不难想到可以用$64$位整数来存$x$。既需要位运算$\sim x$，又需要四则运算$x+1$，不方便进行可合并的线性变换，可以考虑把它们都转化为位运算，或者都转化为四则运算。取反这个位运算不难让人联想到$-x=\sim x+1$的等式（补码性质），所以$\sim x=-x-1$

转化后，可以考虑线段树维护区间线性变化，即区间$[l,r]$的事件会令$x$变为$fx+a$，$A$事件$f=-1,a=-1$，$B$事件$f=1,a=1$。

线段树pushup时，设当前节点为$p$，左儿子为$ls$，右儿子为$rs$，$p$节点的线性变化相当于先进行做儿子的线性变化，再进行右儿子的线性变化，即$x=rs.f\times (ls.f\times x+ls.a)+rs.a=(ls.f\times rs.f)\times x+(rs.f\times ls.a+rs.a)$，是一个简单的区间多信息合并，可以关注一下我的写法，我认为比较简洁。

需要注意一个细节，按我的做法，先把$x$解释成$64$位整数，算出来的$ans$是也是$64$为整数，若$ans$为负，则$ans$的二进制第$len(x)$位到第$63$位是$1$，但根据题意，我们要把$x$重新解释成$len(x)$位无符号整数。

时间复杂度为$O(n\log n+q\log n)$，空间复杂度为$O(n)$

#include<bits/stdc++.h>

using namespace std;
using ll = long long;
const int N = 2e5 + 5;
int n, q;
string s;
struct sgt {
    int l, r;
    int f, a; // x -> f * x + a
} t[N << 2];
#define ls (p << 1)
#define rs (p << 1 | 1)

sgt pushup(const sgt &sl, const sgt &sr) {
    return {sl.l, sr.r, sl.f * sr.f, sl.a * sr.f + sr.a};
}

void build(int p, int l, int r) {
    if (l == r) {
        t[p].l = l;
        t[p].r = r;
        if (s[l] == 'A') {
            t[p].f = -1;
            t[p].a = -1;
        }
        else {
            t[p].f = 1;
            t[p].a = 1;
        }
        return;
    }
    int mid = l + r >> 1;
    build(ls, l, mid);
    build(rs, mid + 1, r);
    t[p] = pushup(t[ls], t[rs]);
}

sgt qry(int p, int l, int r) {
    if (l <= t[p].l && t[p].r <= r) return t[p];
    int mid = t[p].l + t[p].r >> 1;
    if (r <= mid)return qry(ls, l, r);
    if (l > mid)return qry(rs, l, r);
    return pushup(qry(ls, l, r), qry(rs, l, r));
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    cin >> n >> q >> s;
    s = " " + s;
    build(1, 1, n);
    ll lastans = 0;
    while (q--) {
        int tl, tr;
        string strx;
        cin >> tl >> tr >> strx;
        ll x = stoll(strx, nullptr, 2); // string to long long
        int l = (tl ^ lastans) % n + 1;
        int r = (tr ^ lastans) % n + 1;
        if (l > r) swap(l, r);
        sgt c = qry(1, l, r);
        lastans = x * c.f + c.a & (1ll << strx.size()) - 1; // 保留低len位
        for (int i = strx.size() - 1; i >= 0; --i) cout << (lastans >> i & 1);
        cout << "\n";
    }
}