Educational Codeforces Round 23 E. Choosing The Commander Trie

链接：

http://codeforces.com/contest/817/problem/E

题意：

3种操作：

1 插入一个数

2 删除一个数

3 给出一个数pi和l，询问有多少个数pj满足pi^pj<l

题解：

把每个数用bitset变成字符串，然后建字典树，询问的时候当碰到 l 的某一位为1的某一位为1的时候，就更新结果，然后继续搜索

代码：

int Trie[MAXN][2], val[MAXN];
int sz = 1;

void insert(string s, int value) {
    int now = 1;
    rep(i, 0, s.length()) {
        int x = s[i] - '0';
        if (!Trie[now][x]) Trie[now][x] = ++sz;
        now = Trie[now][x];
        val[now] += value;
    }
}

int query(string s, string t) {
    int now = 1, ret = 0;
    rep(i, 0, s.length()) {
        int a = s[i] - '0';
        int b = t[i] - '0';
        if (b) ret += val[Trie[now][a]];
        now = Trie[now][a^b];
    }
    return ret;
}

int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    int q;
    cin >> q;
    while(q--) {
        int k, p, l;
        cin >> k >> p;
        if (k == 1) {
            string s = bitset<30>(p).to_string();
            insert(s, 1);
        }
        else if (k == 2) {
            string s = bitset<30>(p).to_string();
            insert(s, -1);
        }
        else {
            cin >> l;
            string s = bitset<30>(p).to_string();
            string t = bitset<30>(l).to_string();
            cout << query(s, t) << endl;
        }
    }
    return 0;
}