P3292 [SCOI2016]幸运数字(线性基+倍增lca)

题目描述

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
题意:多次询问,每次询问确定一条点到点的路径,确定该路径上选择一些点使其异或最大。

思路:首先想到线性基能找到异或最大,但线性基是线性的,无法在树上使用,要想在树上使用的话就要将树分解成一条条的链,这样就可以使用树链剖分或者lca,但是分解成之后我的结果怎么合并呢?
想不到了,查题解了。
查题解就是简单,每一段的维护的线性基是可以合并的,直接把一个线性基的每一位的元素插入到另一个线性基就可以了,也不会是复杂度高很多。
那么只需要维护每一条链上的线性基,查询的时候将该路径上的所有链的线性基合并就好了,使用lca更简单一点,和祖先节点的标记一起维护线性基就好了。

#include <cstdio>
#include <iostream>
#include <cstring>

using namespace std;
const int maxn=2e4+10;
typedef long long ll;
struct node
{
    int x;
    ll d[61];
    node(){for(int i=0;i<=60;i++) d[i]=0;}
    void add(ll t)
    {
        for(int i=60;i>=0;i--)
        {
            if(t&(1ll<<i))
            {
                if(d[i]==0)
                {
                    d[i]=t;
                    break;
                }
                else t^=d[i];
            }
        }
    }
    ll ans()
    {
        ll ret=0;
        for(int i=60;i>=0;i--)
        {
            if((ret^d[i])>ret) ret^=d[i];
        }
        return ret;
    }
}f[maxn][15];

int id[maxn<<1],nt[maxn<<1],head[maxn<<1],cnt=0,deep[maxn];
ll a[maxn];
int n,q;
void init()
{
    memset(head,-1,sizeof(head));
    cnt=0;
}
void update(int x,int y)
{
    id[cnt]=y;
    nt[cnt]=head[x];
    head[x]=cnt++;
}

void dfs(int u,int pre)
{
    for(int i=head[u];i!=-1;i=nt[i])
    {
        int v=id[i];
        if(v==pre) continue;
        f[v][0].x=u;
        f[v][0].add(a[u]);
        f[v][0].add(a[v]);
        deep[v]=deep[u]+1;
        dfs(v,u);
    }
}

ll work(int x,int y)
{
    node ret;
    if(deep[x]<deep[y]) swap(x,y);
    for(int i=14;i>=0;i--)
    {
        if(deep[f[x][i].x]>=deep[y])
        {
            for(int k=60;k+1;k--)
                if(f[x][i].d[k]) ret.add(f[x][i].d[k]);
            x=f[x][i].x;
        }
    }
    if(x==y) return ret.ans();

    for(int i=14;i+1;i--)
    {
        if(f[x][i].x!=f[y][i].x)
        {
            for(int k=60;k+1;k--)
            {
                if(f[x][i].d[k]) ret.add(f[x][i].d[k]);
                if(f[y][i].d[k]) ret.add(f[y][i].d[k]);
            }
            x=f[x][i].x;
            y=f[y][i].x;
        }
    }

    for(int k=60;k+1;k--)
    {
        if(f[x][0].d[k]) ret.add(f[x][0].d[k]);
        if(f[y][0].d[k]) ret.add(f[y][0].d[k]);
    }

    return ret.ans();
}

int main()
{
    cin>>n>>q;
    init();
    for(int i=1;i<=n;i++)
        scanf("%lld",a+i);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        update(u,v);
        update(v,u);
    }
    deep[1]=1;
    dfs(1,0);

    for(int j=1;j<=14;j++)
        for(int i=1;i<=n;i++)
        {
            f[i][j].x=f[f[i][j-1].x][j-1].x;
            for(int k=60;k+1;k--)
            {
                f[i][j].d[k]=f[i][j-1].d[k];
            }
            for(int k=60;k+1;k--)
            {
                if(f[f[i][j-1].x][j-1].d[k]) f[i][j].add(f[f[i][j-1].x][j-1].d[k]);
            }
        }

    while(q--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        if(x==y) cout<<a[x]<<endl;
        else cout<<work(x,y)<<endl;
    }
}

 

posted @ 2020-12-18 11:04  Npunchman  阅读(90)  评论(0)    收藏  举报