Comet OJ - Contest #12 D.XOR Pair(数位DP)

题目链接

题目描述：

给出4个非负整数a，b，n和m，求出有多少数对$(x,y)(0\le x\le a,0\le y\le b)$满足$x\oplus y=n并且|x-y|\le m$，其中$\oplus$是异或位运算。a,b,n,m<=1e18

解题思路：

如果不管$|x-y|\le m$的限制，这题可以比较容易的想一个数位dp的解法：x和y有上限，且n的每一位可以限制x和y的这一位的取法，具体过程如下：从高位到低位，用lx/ly表示x/y当前是否与a/b相等，并且考虑n对x和y的限制。
但是还要考虑$|x-y|\le m$，这个绝对值看着很碍眼，把它去掉：
$(|x-y|\le m)=>(x-y\le m且y-x\le m)=>(y-x+m\ge 0且x-y+m\ge m)$
变成第三个的时候，只要填当前位的时候只要知道前面填的高位是什么就知道当前哪些可以填，哪些不可以了。但是高位数字的可能性太多了，不可能作为dp数组的某一维。但是注意到这是二进制数字，且y-x+m/x-y+m在当前位填的范围在[-1,2]，所以当高位的数字大于等于1时，后面即使全填-1，也不可能小于0，同样，当高位数字小于等于-2的时候，后面即使全填2也不能让它大于等于0.所以高位的数字只能有-1，0，大于0这三种情况。两个不等式都要满足，所以有两维来表示$|x-y|\le m$的限制。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a, b, n, m;
ll dp[64][3][3];
ll dfs(int pos, int v1, int v2, int lx, int ly){
    v1 = min(v1, 1); v2 = min(v2, 1);
    if(v1 < -1 || v2 < -1) return 0;
    if(pos < 0) return v1 >= 0 && v2 >= 0;
    if(!lx && !ly && ~dp[pos][v1+1][v2+1]) return dp[pos][v1+1][v2+1];
    ll ans = 0;
    int upx = lx ? (a >> pos & 1) : 1;
    int upy = ly ? (b >> pos & 1) : 1;
    for(int x = 0; x <= upx; ++x){
        for(int y = 0; y <= upy; ++y){
            if((x ^ y) != (n>>pos & 1)) continue;
            int t = (m>>pos&1);
            ans += dfs(pos-1, v1*2 + x - y + t, v2*2 + y - x + t, lx && x ==upx, ly && y == upy);
        }
    }
    if(!lx && !ly) dp[pos][v1+1][v2+1] = ans;
    return ans;
}
int main()
{
	int T;cin>>T;
	while(T--){
        memset(dp, -1, sizeof dp);
        scanf("%lld%lld%lld%lld", &a, &b, &n, &m);
        printf("%lld\n",dfs(62, 0, 0, 1, 1) );
	}
}