[BZOJ4568][SCOI2016]幸运数字(倍增LCA,点分治+线性基)

4568: [Scoi2016]幸运数字

Time Limit: 60 Sec  Memory Limit: 256 MB
Submit: 2131  Solved: 865
[Submit][Status][Discuss]

Description

A 国共有 n 座城市，这些城市由 n-1 条道路相连，使得任意两座城市可以互达，且路径唯一。每座城市都有一个

幸运数字，以纪念碑的形式矗立在这座城市的正中心，作为城市的象征。一些旅行者希望游览 A 国。旅行者计划

乘飞机降落在 x 号城市，沿着 x 号城市到 y 号城市之间那条唯一的路径游览，最终从 y 城市起飞离开 A 国。

在经过每一座城市时，游览者就会有机会与这座城市的幸运数字拍照，从而将这份幸运保存到自己身上。然而，幸

运是不能简单叠加的，这一点游览者也十分清楚。他们迷信着幸运数字是以异或的方式保留在自己身上的。例如，

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

有些聪明的游览者发现，只要选择性地进行拍照，便能获得更大的幸运值。例如在上述三个幸运值中，只选择 5 

和 11 ，可以保留的幸运值为 14 。现在，一些游览者找到了聪明的你，希望你帮他们计算出在他们的行程安排中

可以保留的最大幸运值是多少。

Input

第一行包含 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

Output

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

Sample Input

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

Sample Output

14
11

HINT

Source

[Submit][Status][Discuss]

线性基不支持删除，但是支持插入与合并，于是显然可以树剖维护，$O(nlog^4n)$。

线性基不支持修改，浪费了线段树支持修改的功能，实际上可以直接用不支持修改的ST表，$O(nlog^3n)$。

点分治不仅可以做路径统计问题，还可以处理与路径有关的询问问题，将每个询问的两个点的vector中放入这个询问，每次递归到一个重心时查找管辖范围内的所有询问，因为一个询问只涉及两个点，所以复杂度是有保证的。$O(nlog^2n)$

下面是倍增LCA的代码，要注意关于点的LCA和普通的是有区别的：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define rep(i,l,r) for (int i=l; i<=r; i++)
typedef long long ll;
using namespace std;

const int N=20100,M=62;
ll g[N][16][M],Ans[M],a[N];
int n,Q,u,v,cnt,to[N<<1],nxt[N<<1],h[N],d[N],fa[N][16];

void add(int u,int v){ to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt; }
void ins(ll p[],ll x){
    for (int i=61; ~i; i--) if (x&(1ll<<i)){
        if (!p[i]) { p[i]=x; break; } else x^=p[i];
    }
}
void merge(ll g[],ll f1[],ll f2[]){
    rep(i,0,61) g[i]=f1[i];
    rep(i,0,61) if (f2[i]) ins(g,f2[i]);
}
void dfs(int x){
    ins(g[x][0],a[x]);
    rep(i,1,15){
        fa[x][i]=fa[fa[x][i-1]][i-1];
        merge(g[x][i],g[x][i-1],g[fa[x][i-1]][i-1]);
    }
    for (int i=h[x],k; i; i=nxt[i])
        if ((k=to[i])!=fa[x][0]) fa[k][0]=x,d[k]=d[x]+1,dfs(k);
}

void lca(int u,int v){
    memset(Ans,0,sizeof(Ans));
    if (d[u]<d[v]) swap(u,v);
    int t=d[u]-d[v];
    for (int i=15; ~i; i--)
        if (t&(1<<i)) merge(Ans,Ans,g[u][i]),u=fa[u][i];
    if (u==v){ merge(Ans,Ans,g[u][0]); return; }
    for (int i=15; ~i; i--)
        if (fa[u][i]!=fa[v][i])
            merge(Ans,Ans,g[u][i]),merge(Ans,Ans,g[v][i]),
            u=fa[u][i],v=fa[v][i];
    merge(Ans,Ans,g[u][0]); merge(Ans,Ans,g[v][0]);
    merge(Ans,Ans,g[fa[u][0]][0]);
}

ll get(ll p[]){
    ll res=0;
    for (int i=61; ~i; i--) res=max(res,res^p[i]);
    return res;
}

int main(){
     freopen("bzoj4568.in","r",stdin);
     freopen("bzoj45682.out","w",stdout);
    scanf("%d%d",&n,&Q);
    rep(i,1,n) scanf("%lld",&a[i]);
    rep(i,2,n) scanf("%d%d",&u,&v),add(u,v),add(v,u);
    dfs(1);
    while (Q--) scanf("%d%d",&u,&v),lca(u,v),printf("%lld
",get(Ans));
    return 0;
}

然后是点分治，本机时间跑的是倍增的一半，交到OJ上就莫名其妙的死活TLE，弃疗。

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define rep(i,l,r) for (int i=l; i<=r; i++)
#define For(i,x) for (int i=h[x],k; i; i=nxt[i])
typedef long long ll;
using namespace std;

const int N=20010,M=200010,inf=1000000000;
int n,Q,u,v,cnt,S,rt,tim,pos[N],f[N],b[N],sz[N],vis[N],d[N];
int fa[N][16],h[N],to[N<<1],nxt[N<<1];
ll a[N],g[N][65],Ans[65],ans[M];
struct P{ int u,v; }s[M];
vector<int>V[N];
void add(int u,int v){ to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt; }

void ins(ll p[],ll x){
    for (int i=60; ~i; i--) if (x&(1ll<<i)){
        if (!p[i]) { p[i]=x; break; } else x^=p[i];
    }
}

void merge(ll g[],ll f1[],ll f2[]){
    rep(i,0,60) g[i]=f1[i];
    rep(i,0,60) if (f2[i]) ins(g,f2[i]);
}

ll get(ll p[]){
    ll res=0;
    for (int i=60; ~i; i--) res=max(res,res^p[i]);
    return res;
}

void find(int x,int fa){
    f[x]=0; sz[x]=1;
    For(i,x) if ((k=to[i])!=fa && !vis[k])
        find(k,x),sz[x]+=sz[k],f[x]=max(f[x],sz[k]);
    f[x]=max(f[x],S-sz[x]);
    if (f[x]<f[rt]) rt=x;
}

void work(int x,int fa,int bel){
    pos[x]=bel; b[x]=tim;
    rep(i,0,60) g[x][i]=g[fa][i]; ins(g[x],a[x]);
    For(i,x) if ((k=to[i])!=fa && !vis[k]) work(k,x,bel);
}

void work1(int x,int fa){
    for (vector<int>::iterator it=V[x].begin(); it!=V[x].end(); it++){
        int k=*it; if (ans[k]) continue;
        int u=s[k].u; if (u==x) u=s[k].v;
        if (b[u]==tim && (pos[u]!=pos[x] || !pos[u]))
            merge(Ans,g[u],g[x]),ans[k]=get(Ans);
    }
    For(i,x) if ((k=to[i])!=fa && !vis[k]) work1(k,x);
}

void solve(int x){
    vis[x]=1; b[x]=++tim; pos[x]=0;
    rep(i,0,60) g[x][i]=0; ins(g[x],a[x]);
    For(i,x) if (!vis[k=to[i]]) work(k,x,k); work1(x,0);
    For(i,x) if (!vis[k=to[i]]) S=sz[k],f[rt=0]=inf,find(k,x),solve(k);
}

int main(){
    freopen("bzoj4568.in","r",stdin);
    freopen("bzoj4568.out","w",stdout);
    scanf("%d%d",&n,&Q);
    rep(i,1,n) scanf("%lld",&a[i]);
    rep(i,2,n) scanf("%d%d",&u,&v),add(u,v),add(v,u);
    rep(i,1,Q) scanf("%d%d",&s[i].u,&s[i].v),V[s[i].u].push_back(i),V[s[i].v].push_back(i);
    f[rt=0]=inf; S=n; find(1,0); solve(rt);
    rep(i,1,Q) printf("%lld
",ans[i]);
    return 0;
}