bzoj 4568: [Scoi2016]幸運數字【樹鏈剖分+線段樹+線性基】


一眼做法,好處是好想好寫壞處是常數大,容易被卡(bzoj loj 洛谷開O2 能AC,不開有90分……
大概就是樹剖之后維護線段樹,在線段樹的每個節點上上維護一個線性基,暴力\( 60^2 \)的合並兒子
對於每次查詢,在樹上跳重鏈,把這些區間的線性基暴力合並上,然后ans在合並之后的線性基上貪心即可。
這樣,時間復雜度就是預處理\( 60^2nlogn \),查詢的話跳鏈一個log,線段樹查詢60^2log,合並線性基60^2,總的就是\( O(60^2nlog_2n+qlog_2n(60^2log_2n+60^2)) \)

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100005;
int n,q,h[N],cnt,fa[N],de[N],si[N],hs[N],fr[N],id[N],rl[N],tot;
long long a[N];
struct qwe
{
    long long v[65];
    void init()
    {
        memset(v,0,sizeof(v));
    }
    void add(long long x)
    {
        for(int i=60;i>=0;i--)
            if(x>>i)
            {
                if(!v[i])
                {
                    v[i]=x;
                    break;
                }
                x^=v[i];
            }
    }
    qwe operator + (const qwe &a) const
    {
        qwe w=a;
        for(int i=0;i<=60;i++)
            if(v[i])
                w.add(v[i]);
        return w;
    }
};
struct qw
{
    int ne,to;
}e[N<<1];
struct xianduanshu
{
    int l,r;
    qwe v;
}t[N<<1];
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void add(int u,int v)
{
    cnt++;
    e[cnt].ne=h[u];
    e[cnt].to=v;
    h[u]=cnt;
}
void dfs1(int u,int fat)
{
    fa[u]=fat;
    de[u]=de[fat]+1;
    si[u]=1;
    for(int i=h[u];i;i=e[i].ne)
        if(e[i].to!=fat)
        {
            dfs1(e[i].to,u);
            si[u]+=si[e[i].to];
            if(si[e[i].to]>si[hs[u]])
                hs[u]=e[i].to;
        }
}
void dfs2(int u,int top)
{
    fr[u]=top;
    id[u]=++tot;
    rl[tot]=u;
    if(!hs[u])
        return;
    dfs2(hs[u],top);
    for(int i=h[u];i;i=e[i].ne)
        if(e[i].to!=fa[u]&&e[i].to!=hs[u])
            dfs2(e[i].to,e[i].to);
}
void build(int ro,int l,int r)
{
    t[ro].l=l,t[ro].r=r;
    if(l==r)
    {
        t[ro].v.add(a[rl[l]]);
        return;
    }
    int mid=(l+r)>>1;
    build(ro<<1,l,mid);
    build(ro<<1|1,mid+1,r);
    t[ro].v=t[ro<<1].v+t[ro<<1|1].v;
}
qwe ques(int ro,int l,int r)
{
    if(t[ro].l==l&&t[ro].r==r)
        return t[ro].v;
    int mid=(t[ro].l+t[ro].r)>>1;
    if(r<=mid)
        return ques(ro<<1,l,r);
    else if(l>mid)
        return ques(ro<<1|1,l,r);
    else
        return ques(ro<<1,l,mid)+ques(ro<<1|1,mid+1,r);
}
void wen(int u,int v)
{
    qwe w;
    w.init();
    while(fr[u]!=fr[v])
    {
        if(de[fr[u]]<de[fr[v]])
            swap(u,v);
        w=w+ques(1,id[fr[u]],id[u]);
        u=fa[fr[u]];
    }
    if(de[u]>de[v])
        swap(u,v);
    w=w+ques(1,id[u],id[v]);
    long long ans=0;
    for(int i=60;i>=0;i--)
        if((ans^w.v[i])>ans)
            ans^=w.v[i];
    printf("%lld\n",ans);
}
int main()
{
    n=read(),q=read();
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=1;i<n;i++)
    {
        int x=read(),y=read();
        add(x,y),add(y,x);
    }
    dfs1(1,0);
    dfs2(1,1);
    build(1,1,n);
    while(q--)
    {
        int x=read(),y=read();
        wen(x,y);
    }
    return 0;
}

注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
粤ICP备14056181号  © 2014-2020 ITdaan.com