[FZOJ194] 「2019冬令营提高组」密文（Trie+贪心）

题意

• 有一个序列$\mathrm{a}$，你可以花费${\mathrm{v}}_{\mathrm{l}}\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{v}}_{\mathrm{l}+1}\mathrm{x}\mathrm{o}\mathrm{r}...\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{v}}_{\mathrm{r}-1}\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{v}}_{\mathrm{r}}$的代价询问${\mathrm{a}}_{\mathrm{l}}\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{a}}_{\mathrm{l}+1}\mathrm{x}\mathrm{o}\mathrm{r}...\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{a}}_{\mathrm{r}-1}\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{a}}_{\mathrm{r}}$，求确定序列$\mathrm{a}$的最小代价。（$\mathrm{n}\le 10^5$）

知道序列$\mathrm{a}$等价于知道了$\forall \mathrm{i}\in [0,\mathrm{n}]$，${\mathrm{s}}_{\mathrm{i}}={\mathrm{a}}_1\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{a}}_2\mathrm{x}\mathrm{o}\mathrm{r}...{\mathrm{a}}_{\mathrm{i}-1}\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{a}}_{\mathrm{i}}$。那么每次询问$[\mathrm{l},\mathrm{r}]$就相当于知道了${\mathrm{s}}_{\mathrm{l}-1}\mathrm{x}\mathrm{o}\mathrm{r}{\mathrm{s}}_{\mathrm{r}}$，我们可以对任意两个${\mathrm{s}}_{\mathrm{i}}$连询问他们两个的代价的边，因为确定$\mathrm{n}$个未知数需要$\mathrm{n}$个方程，那么这个张图的最小生成树的权值和就是确定序列的最小代价（已知${\mathrm{s}}_0=0）$。

我们考虑克鲁斯卡尔的过程，每次找到当前权值最小的边，连接两个连通块，因为这里的运算为异或，我们用一个Trie来寻找当前权值最小的边，边的总数是$\mathrm{O}({\mathrm{n}}^2)$级别的，但实际上我们按照Trie自底向上连边的话，任意两个点所连的边只会在他们的LCA处作为连接LCA左右两个子树之间两个连通块的边，我们按照启发式合并的思想枚举子树大小较小的所有点，在另一边所在的Trie上查询异或最小值，复杂度就是$\mathrm{O}(\mathrm{n}\log \mathrm{n}\log 10^9)$。

#include <bits/stdc++.h>

using namespace std;

const int N = 1e7 + 3;

struct Trie {
    int ch[N][2], size[N], dis[N], val[N], cnt;

    void insert(int x) {
        int now = 0;
        for (int i = 30; ~i; --i) {
            int to = x >> i & 1;
            if (!ch[now][to])
                ch[now][to] = ++cnt;
            now = ch[now][to], ++size[now];
        }
    }

    queue<int> q;

    int query(int s, int t, int d) {
        if (!size[s] || !size[t])
            return 0;
        if (size[s] > size[t])
            swap(s, t);

        int ans = INT_MAX;

        for (q.push(s); !q.empty(); q.pop()) {
            int u = q.front();
            if (dis[u] > d) {
                int now = t;
                for (int i = d; ~i; --i) {
                    int to = val[u] >> i & 1;
                    if (!ch[now][to])
                        to ^= 1;
                    now = ch[now][to], val[u] ^= to << i;
                }
                ans = min(ans, val[u] | (1 << (d + 1)));
            } else {
                for (int j = 0; j < 2; ++j) {
                    if (size[ch[u][j]]) {
                        val[ch[u][j]] = val[u] | (j << (d - dis[u]));
                        dis[ch[u][j]] = dis[u] + 1, q.push(ch[u][j]);
                    }
                }
            }
            dis[u] = val[u] = 0;
        }
        return ans;
    }

    void dfs(int now, int d) {
        static long long ans = 0;
        if (size[ch[now][0]] > 1)
            dfs(ch[now][0], d - 1);
        if (size[ch[now][1]] > 1)
            dfs(ch[now][1], d - 1);
        ans += query(ch[now][0], ch[now][1], d - 1);
        if (!now)
            printf("%lld\n", ans);
    }

} T;

int n, s[N];

int main() {
    freopen("secret.in", "r", stdin);
    freopen("secret.out", "w", stdout);

    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &s[i]);
        s[i] ^= s[i - 1];
        T.insert(s[i]);
    }
    T.insert(0), T.dfs(0, 30);

    return 0;
}