bzoj 4568: [Scoi2016]幸运数字

ST表维护线性基,用ST表维护不带修改的东西可以比树剖少一个log。注意维护线性基是60*60的复杂度。
    
    
    
    
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
 
#define ll long long
#define inf 1e9
#define eps 1e-8
#define md
#define N 20010
using namespace std;
struct yts { int x,t,ne;} e[2*N];
int fa[N][15],v[N],dep[N];
ll c[N][15][60],Ans[60];
int n,num;
 
void put(int x,int y)
{
num++; e[num].x=x; e[num].t=y;
e[num].ne=v[x]; v[x]=num;
}
 
void dfs(int x)
{
for (int i=v[x];i;i=e[i].ne)
{
int y=e[i].t;
if (y!=fa[x][0])
{
dep[y]=dep[x]+1; fa[y][0]=x;
dfs(y);
}
}
}
 
void merge(ll *ans,ll *a)
{
for (int i=59;i>=0;i--)
if (a[i])
{
ll x=a[i];
for (int j=59;j>=0;j--)
if (x&(1ll<<j))
{
if (ans[j]) x^=ans[j];
else { ans[j]=x; break;}
}
}
}
 
void build_lca()
{
for (int j=1;j<15;j++)
for (int i=1;i<=n;i++)
if (fa[i][j-1])
{
fa[i][j]=fa[fa[i][j-1]][j-1];
memcpy(c[i][j],c[i][j-1],sizeof(c[i][j-1]));
merge(c[i][j],c[fa[i][j-1]][j-1]);
}
}
int get(int x,int y)
{
memset(Ans,0,sizeof(Ans));
if (dep[x]<dep[y]) swap(x,y);
int t=dep[x]-dep[y];
for (int j=14;j>=0;j--)
if (t&(1<<j))
{
merge(Ans,c[x][j]);
x=fa[x][j];
}
if (x==y) return x;
for (int j=14;j>=0;j--)
if (fa[x][j]!=fa[y][j])
{
merge(Ans,c[x][j]); merge(Ans,c[y][j]);
x=fa[x][j]; y=fa[y][j];
}
merge(Ans,c[x][0]); merge(Ans,c[y][0]);
return fa[x][0];
}
int main()
{
int Q;
scanf("%d%d",&n,&Q);
for (int i=1;i<=n;i++)
{
ll x;
scanf("%lld",&x);
Ans[0]=x; merge(c[i][0],Ans);
}
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
put(x,y); put(y,x);
}
dep[1]=1; dfs(1); build_lca();
for (int i=1;i<=Q;i++)
{
int x,y;
scanf("%d%d",&x,&y);
int z=get(x,y);
merge(Ans,c[z][0]);
ll ans=0;
for (int j=59;j>=0;j--)
ans=max(ans,ans^Ans[j]);
printf("%lld\n",ans);
}
return 0;
}


你可能感兴趣的:(bzoj 4568: [Scoi2016]幸运数字)