【jzoj4598】【准备食物】【字典树】

题目大意

给出长度为n的序列，询问[1,r]的后缀区间[i,r] (1<=i,i<=r)异或和大于等于k的区间的个数。

题解

异或可逆，[i,r]等于[1,i-1]异或[1,r]，设前缀和s，求满足s[i-1]^s[r]>=k,由于不可移项，必须考虑s[r]和k的影响。考虑建立Trie在上面查询答案。考虑当先选0还是1，选了一个异或后比k大则所有答案可以贡献，当前相等则继续比较，小于则放弃。详细见code。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const maxn=100000,maxq=100000;
int n,q,pon=1,ans[maxq+10],b[40],c[40],size[maxn*40+10],son[maxn*40+10][2];
long long a[maxn+10];
struct rec{int r,k,num;};
rec qu[maxq+10];
bool cmp(rec i,rec j){
    return (i.r<j.r)||((i.r==j.r)&&(i.num==j.num));
}
void insert(int now,int pos){
    size[now]++;
    if(pos>30)return;
    if(!son[now][b[pos]])son[now][b[pos]]=++pon;
    insert(son[now][b[pos]],pos+1);
}
int count(int now,int pos){
    if(!now)return 0;
    if(pos>30)return size[now];
    if(b[pos]&&c[pos])return count(son[now][0],pos+1);
    else if(b[pos]&&(!c[pos]))return count(son[now][1],pos+1)+size[son[now][0]];
    else if((!b[pos])&&c[pos])return count(son[now][1],pos+1);
    else return count(son[now][0],pos+1)+size[son[now][1]];
}
int main(){
    //freopen("food.in","r",stdin);
    //freopen("food.out","w",stdout);
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n)
        scanf("%lld",&a[i]);
    scanf("%d",&q);
    fo(i,1,q)
        scanf("%d%d",&qu[i].r,&qu[i].k),qu[i].num=i;
    sort(qu+1,qu+q+1,cmp);
    int tmp=0;insert(1,1);
    fo(i,1,q){
        fo(j,qu[i-1].r+1,qu[i].r){
            tmp^=a[j];int tmpp=tmp;
            fd(k,30,1){
                b[k]=tmpp%2;
                tmpp/=2;
            }
            insert(1,1);
        }
        fd(j,30,1){
            c[j]=qu[i].k%2;
            qu[i].k/=2;
        }
        ans[qu[i].num]=count(1,1);
    }
    fo(i,1,q)printf("%d\n",ans[i]);
    return 0;
}