luoguP3292 [SCOI2016]幸运数字（线性基+树上倍增）

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const int maxn=2e4+10;
 5 struct node{
 6     int to,nxt;
 7 }edge[maxn*2];
 8 int head[maxn],f[maxn][20],deep[maxn],tot,N,n,q;
 9 ll a[maxn];
10 struct Linear_basis{
11     ll b[65];
12     bool insert(ll x){
13         for (int i=62;i>=0;i--){
14             if (x&(1ll<<i)){
15                 if (!b[i]){b[i]=x; break;} 
16                 x^=b[i];
17             } 
18         }
19         return x>0;  //是否可插入 true：可插入，false：不可插入 
20     }
21     ll mx(){
22         ll res=0;
23         for (int i=62;i>=0;i--) res=max(res,res^b[i]);
24         return res;
25     }
26     void clear(){
27         memset(b,0,sizeof(b));
28     }
29 }LB[maxn][20],ans;
30 void add(int x,int y){
31     edge[++tot]={y,head[x]}; head[x]=tot;
32 }
33 void dfs(int x,int fa){
34     f[x][0]=fa; //倍增数组 
35     LB[x][0].insert(a[x]); 
36     for (int i=head[x];i!=-1;i=edge[i].nxt){
37         node tmp=edge[i];
38         if (tmp.to==fa) continue;
39         deep[tmp.to]=deep[x]+1;
40         dfs(tmp.to,x);
41     }
42 }
43 Linear_basis merge(Linear_basis p,Linear_basis q){
44     Linear_basis tmp=p;
45     for (int i=62;i>=0;i--){
46         if (q.b[i]) tmp.insert(q.b[i]);
47     }
48     return tmp;
49 }
50 void init(){
51     N=floor(log(1.0*n)/log(2.0));
52     deep[1]=0;
53     dfs(1,0);
54     for (int j=1;j<=N;j++){
55         for (int i=1;i<=n;i++){
56             f[i][j]=f[f[i][j-1]][j-1];
57             LB[i][j]=merge(LB[i][j-1],LB[f[i][j-1]][j-1]); 
58         }
59     }
60 }
61 int _LCA(int x,int y){
62     if (deep[x]<deep[y]) swap(x,y);
63     for (int j=N;j>=0;j--)
64         if (deep[f[x][j]]>=deep[y]){
65             ans=merge(ans,LB[x][j]);
66             x=f[x][j];
67         }
68     if (x==y){
69         ans=merge(ans,LB[x][0]); return x;
70     }
71     for (int j=N;j>=0;j--){
72         if (f[x][j]!=f[y][j]){
73             ans=merge(ans,merge(LB[x][j],LB[y][j]));
74             x=f[x][j]; y=f[y][j];
75         }
76     }
77     Linear_basis tmp=merge(LB[x][0],LB[y][0]);
78     Linear_basis tmp2=merge(ans,LB[f[x][0]][0]);
79     ans=merge(tmp2,tmp);
80     return f[x][0]; //lca
81 }
82 int main(){
83     scanf("%d%d",&n,&q);
84     for (int i=1;i<=n;i++) scanf("%lld",&a[i]),head[i]=-1;
85     int x,y; tot=0;
86     for (int i=1;i<=n-1;i++){
87         scanf("%d%d",&x,&y);
88         add(x,y);add(y,x);
89     }
90     init();
91     while (q--){
92         scanf("%d%d",&x,&y);
93         ans.clear(); 
94         _LCA(x,y);
95         printf("%lld\n",ans.mx()); 
96     }
97     return 0;
98 }