洛谷P3292 [SCOI2016]幸运数字 线性基+倍增

P3292 [SCOI2016]幸运数字

传送门

题目描述

A 国共有 n 座城市，这些城市由 n-1 条道路相连，使得任意两座城市可以互达，且路径唯一。每座城市都有一个幸运数字，以纪念碑的形式矗立在这座城市的正中心，作为城市的象征。

一些旅行者希望游览 A 国。旅行者计划乘飞机降落在 x 号城市，沿着 x 号城市到 y 号城市之间那条唯一的路径游览，最终从 y 城市起飞离开 A 国。在经过每一座城市时，游览者就会有机会与这座城市的幸运数字拍照，从而将这份幸运保存到自己身上。然而，幸运是不能简单叠加的，这一点游览者也十分清楚。他们迷信着幸运数字是以异或的方式保留在自己身上的。

例如，游览者拍了 3 张照片，幸运值分别是 5，7，11，那么最终保留在自己身上的幸运值就是 9（5 xor 7 xor 11）。

有些聪明的游览者发现，只要选择性地进行拍照，便能获得更大的幸运值。例如在上述三个幸运值中，只选择 5 和 11 ，可以保留的幸运值为 14 。现在，一些游览者找到了聪明的你，希望你帮他们计算出在他们的行程安排中可以保留的最大幸运值是多少。

输入格式

第一行包含 2 个正整数 n ，q，分别表示城市的数量和旅行者数量。

第二行包含 n 个非负整数，其中第 i 个整数 Gi 表示 i 号城市的幸运值。

随后 n-1 行，每行包含两个正整数 x ，y，表示 x 号城市和 y 号城市之间有一条道路相连。

随后 q 行，每行包含两个正整数 x ，y，表示这名旅行者的旅行计划是从 x 号城市到 y 号城市。N<=20000,Q<=200000,Gi<=2^60

输出格式

输出需要包含 q 行，每行包含 1 个非负整数，表示这名旅行者可以保留的最大幸运值。

输入输出样例

输入

4 2
11 5 7 9
1 2
1 3
1 4
2 3
1 4

输出

14
11

题解：要求一条路径上的最大异或值，我们知道求最大异或值可以通过线性基来做，那么我们怎么得到两个城市之间的线性基呢？我们可以知道这n个城市及其道路构成了一棵树，我们可以通过找这两个城市的LCA，再将这两个城市到其LCA的两个线性基合并起来得到这路径上的LCA。把平时倍增用的dis数组改为记录线性基的数组维护。
代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e4 + ;
ll a[N],b[N][][],dep[N],ans[],fa[N][];
vector<int> v[N];
ll read() {
    ll x = , f = ;
    char ch = getchar();
    while(ch>''||ch<''){
        if (ch == '-') f = -;
        ch = getchar();
    }
    while(ch >= '' && ch <= '') {
        x = x *  + ch-'';
        ch = getchar();
    }
    return x * f;
}
void add(ll B[],ll x) {
    for (int i = ; i >= ; i--) {
        if (x & (1ll<<i)) {
            if (B[i]) x^=B[i];
            else {
                B[i] = x;
                break;
            }
        }
    }
}
void merge(ll B1[],ll B2[]){
    for (int i = ; i >= ; i--)
        if (B2[i]) add(B1,B2[i]);
}
void dfs(int f,int u) {
    fa[u][] = f;
    dep[u] = dep[f] + ;
    for (int i = ; i <= ; i++) {
        fa[u][i] = fa[fa[u][i-]][i-];
        merge(b[u][i],b[u][i-]);
        merge(b[u][i] ,b[fa[u][i-]][i-]);
    }
    for (int i = ; i < v[u].size(); i++) {
        if (f == v[u][i]) continue;
        dfs(u,v[u][i]);
    }
}
void lca(int x,int y) {
    memset(ans,,sizeof(ans));
    if (dep[x] < dep[y]) swap(x,y);
    for (int i = ; i >= ; i--)
        if (dep[fa[x][i]] >= dep[y]) {
            merge(ans,b[x][i]);
            x = fa[x][i];
        }
    if (x == y) {
        merge(ans,b[x][]);
        return;
    }
    for (int i = ; i >= ; i--)
        if (fa[x][i] != fa[y][i]) {
            merge(ans,b[x][i]);
            merge(ans,b[y][i]);
            x = fa[x][i];
            y = fa[y][i];
        }
    merge(ans,b[x][]);
    merge(ans,b[y][]);
    merge(ans,b[fa[x][]][]);
}
int main() {
    ll n = read(),q = read(),x,y;
    for (ll i = ; i <= n; i++)
        add(b[i][],a[i] = read());
    for(ll i = ;i < n; i++){
        x = read(),y = read();
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(,);
    while(q--) {
        x = read(),y = read();
        lca(x,y);
        ll sum = ;
        for (int i = ; i >= ; i--)
            sum = max(sum,sum^ans[i]);
        printf("%lld\n", sum);
    }
    return ;
}

//变量写错找bug找了一个多小时