AtCoder Regular Contest 133 D.Range XOR(数位dp+分类讨论)

题目

给定L,R,V(1<=L<=R<=1e18,0<=V<=1e18)

求满足L<=l<=r<=R，且(l,r)之间的数的异或和恰等于v的(l,r)二元组个数，

答案对998244353取模

思路来源

YLWang代码

题解

heltion的dp把各个情况都统一了，但是看不懂orz，就补这个吧

官方题解，也是数位dp+分类讨论

注意到[0,x]的前缀异或取值只有四种，

①x%4==0时，为x

②x%4==1时，为1

③x%4==2时，为x+1

④x%4==3时，为0

[l,r]异或和，即前缀异或和sum[r]^sum[l-1]

于是分l-1模4的值，r模4的值，16种情况讨论

比较难处理的是以下四种情况：

①(l-1)%4=0且r%4=0

②(l-1)%4=0且r%4=2

③(l-1)%4=2且r%4=0

④(l-1)%4=2且r%4=2

其余12种情况，

要么可以确定一个端点的位置，

要么形如等差数列，均可快速求得

这4种情况，可以归一到一个问题上，

即给定区间[x,y]，从区间内抽取两个不相等的数lb,rb(lb<rb)，求lb^rb=w的二元组对数(lb,rb)

这里的x实际是l-1，而y仍然是r

因此，引入数位dp，

dp[i][j][k]表示还有i位，第一个数是/否卡上界，第二个数是否卡上界的方案数

一维数位dp，一般用dfs(r)-dfs(l-1)，前缀相减求答案

二维也同理，由于其只能控制卡不卡上界，所以卡下界的问题交给容斥，

类似二维矩阵求(l,l)到(r,r)的面积，用总的面积减两个矩形面积，再加回左上角矩形的面积

类比可知，最后二维的答案为solve(r,r)-2*solve(l-1,r)+solve(l-1,l-1)

由于实际问题中l-1<r，所以当v=0时，需要把任意抽取的相等的数(l-1==r)的数减掉，

而且由于数位dp中，i和j是对称的，没有大小关系的，所以最终答案再除以2

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10,mod=998244353,inv2=(mod+1)/2;
ll l,r,v;
int ans,a[66],b[66],dp[66][2][2];
void add(int &x,int y){
    x+=y;
    if(x>=mod)x-=mod;
    if(x<0)x+=mod;
}
int dfs(int pos,bool lima,bool limb,ll my_v){
	if(pos==-1){
        return 1;
	}
	if(~dp[pos][lima][limb])return dp[pos][lima][limb];
    int ret=0;
    int upa=lima?a[pos]:1,upb=limb?b[pos]:1;
    for(int i=0;i<=upa;++i){
        for(int j=0;j<=upb;++j){
            if(pos==0 && (i || j))continue;//l:(0,2) r:(0,2)
            if(pos==1){
                if((i^j) && (v%4!=3))continue;//l:0 r:2; l:2 r:0
                if((i==j) && (v%4!=0))continue;//l:0 r:0; l:2 r:2
            }
            if((i^j) != (my_v>>pos&1))continue;
            add(ret,dfs(pos-1,lima && i==upa,limb && j==upb,my_v));
        }
    }
    dp[pos][lima][limb]=ret;
    return ret;
}
int solve(ll x,ll y,ll v){
    memset(a,0,sizeof a);
    memset(b,0,sizeof b);
	int sza=0,szb=0;
	for(;x;x/=2)a[sza++]=x%2;
    for(;y;y/=2)b[szb++]=y%2;
    memset(dp,-1,sizeof dp);
    return dfs(63,1,1,v);
}
int cal(ll lb,ll rb,int re){
    if(rb<re || lb>rb)return 0;
    ll nowl=0,nowr=0;
    nowr=(rb-re)/4+1;
    lb--;
    if(lb<re)nowl=0;
    else nowl=(lb-re)/4+1;
    ll ans=nowr-nowl;
    ans%=mod;
    return (int)ans;
}
//[lb,rb]内取l<=r,使得(l^r)==v的数对数,特判lb=0
int cal2(ll lb,ll rb,ll v){
    int ans=0;
    ans=((1ll*solve(rb,rb,v)-2ll*solve(lb-1,rb,v)+solve(lb-1,lb-1,v))%mod+mod)%mod;
    if(!v){//l=r不合法,l是实际的l-1
        ans=(ans-cal(lb,rb,0)+mod)%mod;
        ans=(ans-cal(lb,rb,2)+mod)%mod;
    }
    ans=1ll*ans*inv2%mod;
    return ans;
}
int main(){
    scanf("%lld%lld%lld",&l,&r,&v);
    if(v==0){
        int three=cal(l-1,r,3);//l-1:3 r:3 C(x,2)
        add(ans,1ll*(three-1)*three%mod*inv2%mod);
        int one=cal(l-1,r,1);//l-1:1 r:1 C(x,2)
        add(ans,1ll*(one-1)*one%mod*inv2%mod);
    }
    if(v==1){
        int two=cal(l,r,2);
        if(two){//l-1:1 r:3
            ll ltwo=l;while(ltwo%4!=2)ltwo++;
            int lthree=cal(ltwo,r,3);
            ll rtwo=r;while(rtwo%4!=2)rtwo--;
            int rthree=cal(rtwo,r,3);
            add(ans,1ll*(lthree+rthree)*two%mod*inv2%mod);
        }
        int zero=cal(l,r,0);
        if(zero){//l-1:3 r:1
            ll lzero=l;while(lzero%4)lzero++;
            int lone=cal(lzero,r,1);
            ll rzero=r;while(rzero%4)rzero--;
            int rone=cal(rzero,r,1);
            add(ans,1ll*(lone+rone)*zero%mod*inv2%mod);
        }
    }
    if(v%4==0){
        if(l<=v+1 && v+1<=r){
            add(ans,cal(v+1,r,3));//l-1:0 r:3
        }
        if(l<=v && v<=r){
            add(ans,cal(l,v,0));//l-1:3 r:0
        }
        if(l>1)add(ans,cal2(l-1,r,v));//l-1:0 r:0 l-1:2 r:2
        else add(ans,cal2(1,r,v)+(l<=v && v<=r));//特判l-1=0,拆成两部分统计
    }
    if(v%4==1){
        if(l<=v && v<=r){
            add(ans,cal(v,r,1));//l-1:0 r:1
        }
        if(l<=(v^1) && (v^1)<=r){
            add(ans,cal(l,v^1,2));//l-1:1 r:0
        }
    }
    if(v%4==2){
        if(l<=v && v<=r){
            add(ans,cal(l,v,2));//l-1:1 r:2
        }
        if(l<=(v^1) && (v^1)<=r){
            add(ans,cal(v^1,r,1));//l-1:2 r:1
        }
    }
    if(v%4==3){
        if(l<=v && v<=r){
            add(ans,cal(v,r,3));//l-1:2 r:3
        }
        if(l<=(v^1) && (v^1)<=r){
            add(ans,cal(l,v^1,0));//l-1:3 r:2
        }
        if(l>1)add(ans,cal2(l-1,r,v^1));//l-1:0 r:2;l-1:2 r:0
        else add(ans,cal2(1,r,v^1)+(l<=(v^1) && (v^1)<=r));//特判l-1=0,拆成两部分统计
    }
    printf("%d\n",ans);
    return 0;
}