CF1055F Tree and XOR

CF1055F Tree and XOR

就是选择两个数找第k大对儿

第k大？二分+trie上验证

O(nlognlogn)

直接按位贪心

维护可能的决策点(a,b)表示可能答案的对儿在a和b的子树中选择

所有可能决策点都贡献这一位是0，看是否<=k

然后更新出下一层的决策点

但是空间太小，

所以要滚动

我的方法：

维护trie节点和控制区间，维护每个区间的元素，维护决策点

注意，(a,b)(b,a)算两个。考虑a!=b时候贡献*=2

#include<bits/stdc++.h>
#define reg register int
#define il inline
#define fi first
#define se second
#define mk(a,b) make_pair(a,b)
#define numb (ch^'0')
#define pb push_back
#define solid const auto &
#define enter cout<<endl
#define pii pair<int,int>
// #define int long long
using namespace std;
typedef long long ll;
template<class T>il void rd(T &x){
    char ch;x=;bool fl=false;
    while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);
    for(x=numb;isdigit(ch=getchar());x=x*+numb);
    (fl==true)&&(x=-x);
}
template<class T>il void output(T x){if(x/)output(x/);putchar(x%+'');}
template<class T>il void ot(T x){if(x<) putchar('-'),x=-x;output(x);putchar(' ');}
template<class T>il void prt(T a[],int st,int nd){for(reg i=st;i<=nd;++i) ot(a[i]);putchar('\n');}

namespace Miracle{
const int N=1e6+;
ll k,a[N];
int b[][N],num[];
ll ans;
struct tr{
    int ls,rs;
    int l,r;
    void init(){
        ls=rs=l=r=;
    }
    void up(int ll,int rr){
        l=ll;r=rr;
    }
    int sz(){
        if(l&&r) return r-l+;
        return ;
    }
}t[][N];
int tot[];
int n;
pii s[][N];
int q[];
struct node{
    int nxt,to;
    ll val;
}e[N];
int hd[N],cnt;
void add(int x,int y,ll z){
    e[++cnt].nxt=hd[x];
    e[cnt].to=y;e[cnt].val=z;
    hd[x]=cnt;
}
void dfs(int x,ll dis){
    b[][++num[]]=x;a[x]=dis;
    for(reg i=hd[x];i;i=e[i].nxt){
        int y=e[i].to;
        dfs(y,dis^e[i].val);
    }
}
int tmp;
void build(int x,int d){
    int las=tmp,now=tmp^;
    int bc=t[las][x].r;
    for(reg i=t[las][x].l;i<=t[las][x].r;++i){
        if(!(a[b[las][i]]&(1LL<<d))){
            b[now][++num[now]]=b[las][i];
        }else{
            b[now][bc--]=b[las][i];
        }
    }
    if(num[now]>=t[las][x].l){
        ++tot[now];
        t[las][x].ls=tot[now];
        t[now][tot[now]].init();
        t[now][tot[now]].up(t[las][x].l,num[now]);
    }
    if(num[now]<t[las][x].r){
        ++tot[now];
        t[las][x].rs=tot[now];
        t[now][tot[now]].init();
        t[now][tot[now]].up(num[now]+,t[las][x].r);
    }
    num[now]=t[las][x].r;
}
int main(){
    rd(n);rd(k);
    int y;ll w;
    for(reg i=;i<=n;++i){
        rd(y);rd(w);add(y,i,w);
    }
    dfs(,);
    t[][++tot[]].up(,num[]);
    num[]=;tot[]=;
    build(tot[],);
    s[][++q[]]=mk(,);
    for(reg d=;d>=;--d){
        ll con=;
        int las=tmp;
        int now=tmp^;
        for(reg i=;i<=q[las];++i){
            if(s[las][i].fi==s[las][i].se){
                con+=(ll)t[now][t[las][s[las][i].fi].ls].sz()*t[now][t[las][s[las][i].se].ls].sz()+
                (ll)t[now][t[las][s[las][i].fi].rs].sz()*t[now][t[las][s[las][i].se].rs].sz();
            }else{
                con+=(ll)t[now][t[las][s[las][i].fi].ls].sz()*t[now][t[las][s[las][i].se].ls].sz()*+
                (ll)t[now][t[las][s[las][i].fi].rs].sz()*t[now][t[las][s[las][i].se].rs].sz()*;
            }
        }
        // cout<<" con "<<con<<endl;
        if(k>con){//
            k-=con;
            ans+=(1LL<<d);
            q[now]=;
            for(reg i=;i<=q[las];++i){
                if(t[las][s[las][i].fi].ls&&t[las][s[las][i].se].rs){
                    s[now][++q[now]]=mk(t[las][s[las][i].fi].ls,t[las][s[las][i].se].rs);
                }
                if(s[las][i].fi!=s[las][i].se){
                    if(t[las][s[las][i].fi].rs&&t[las][s[las][i].se].ls){
                        s[now][++q[now]]=mk(t[las][s[las][i].fi].rs,t[las][s[las][i].se].ls);
                    }
                }
            }
        }else{
            q[now]=;
            for(reg i=;i<=q[las];++i){
                if(t[las][s[las][i].fi].ls&&t[las][s[las][i].se].ls){
                    s[now][++q[now]]=mk(t[las][s[las][i].fi].ls,t[las][s[las][i].se].ls);
                }
                if(t[las][s[las][i].fi].rs&&t[las][s[las][i].se].rs){
                    s[now][++q[now]]=mk(t[las][s[las][i].fi].rs,t[las][s[las][i].se].rs);
                }
            }
        }
        tmp^=;
        if(d){
            swap(now,las);
            tot[now]=;
            num[now]=;
            for(reg i=;i<=tot[las];++i){
                build(i,d-);
            }
        }
    }
    ot(ans);
    return ;
}   

}
signed main(){
    Miracle::main();
    return ;
}

/*
   Author: *Miracle*
*/

然后这个代码又臭又长

一个很nb的写法：

1.首先不用建树，p<=i，直接v[i]=v[p]^w，一行搞定

2.不用维护决策两个点，只用维护每个元素，可能匹配的子树节点位置！以及自己的权值属于的位置

3.每次更新节点的size和a，s和k比较大小

4.更新b，

也根本不用0/1滚动，b和a数组已经区分了位置。

#include<bits/stdc++.h>
#define reg register int
#define il inline
#define fi first
#define se second
#define mk(a,b) make_pair(a,b)
#define numb (ch^'0')
#define pb push_back
#define solid const auto &
#define enter cout<<endl
#define pii pair<int,int>
using namespace std;
typedef long long ll;
template<class T>il void rd(T &x){
    char ch;x=;bool fl=false;
    while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);
    for(x=numb;isdigit(ch=getchar());x=x*+numb);
    (fl==true)&&(x=-x);
}
template<class T>il void output(T x){if(x/)output(x/);putchar(x%+'');}
template<class T>il void ot(T x){if(x<) putchar('-'),x=-x;output(x);putchar(' ');}
template<class T>il void prt(T a[],int st,int nd){for(reg i=st;i<=nd;++i) ot(a[i]);putchar('\n');}

namespace Miracle{
const int N=1e6+;
ll k,s,t,v[N];
int n;
int a[N],b[N],ch[N][],sz[N];
int tot;
ll ans;
int get(int x,int c){
    return ch[x][c]?ch[x][c]:ch[x][c]=++tot;
}
int main(){
    rd(n);rd(k);int p;ll w;
    for(reg i=;i<=n;++i) rd(p),rd(w),v[i]=v[p]^w;
    for(reg i=;i<=n;++i) a[i]=b[i]=;
    for(reg d=;d>=;--d){
        for(reg i=;i<=tot;++i) ch[i][]=ch[i][]=sz[i]=;
        tot=s=t=;
        for(reg i=;i<=n;++i) sz[a[i]=get(a[i],v[i]>>d&)]++;
        for(reg i=;i<=n;++i) s+=sz[ch[b[i]][v[i]>>d&]];
        if(s<k) k-=s,t=,ans|=(1LL<<d);
        for(reg i=;i<=n;++i) b[i]=ch[b[i]][(v[i]>>d&)^t];
    }
    ot(ans);
    return ;
}

}
signed main(){
    Miracle::main();
    return ;
}

/*
   Author: *Miracle*
*/

tql！