ABC50 Xor Sum

Xor Sum

题目问的是有多少对u和v，大小在[0,n]之间使得存在a和b，满足a xor b=u 并且a + b=v

首先能够想到的是若要满足a+b=v<=n，必须满足a xor b=a+b-((a&b)<<1)<=n，所以只需要考虑a+b=v<=n，a xor b自然也是一个小于等于n的值，设a=(a1*2 + a2), b = (b1*2+b2),a2和b2分别是a和b转换为二进制下的最低位，a1和b1是a和b转换为二进制下的最低位为0的值，这样就可以得到a1*2+a2+b1*2+b2=v<=n，易知a2+b2有三种情况，分别等于0，1，2，所以a1*2+b1*2<=(n-a2-b2)，->a1+b1<=(n-a2-b2)/2，即n中解的个数分别来自于(n/2)解的个数+((n-1)/2)解的个数+((n-2)/2)解的个数，可知当n=0，结果为1，当n=1，结果为2，所以可以记忆化搜索解决

AC代码：

#include <bits/stdc++.h>
using namespace std;
using LL = long long;
const LL mod = 1e9 + 7;
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    LL n;
    cin >> n;
    map<LL, LL> mp;
    mp[0] = 1;
    mp[1] = 2;
    function<LL(LL)> sol = [&](LL a) {
        if (mp[a]) {
            return mp[a] % mod;
        }
        return mp[a] = (((LL)sol(a / 2) + sol((a - 1) / 2)) % mod + sol((a - 2) / 2)) % mod;
    };
    cout << sol(n) << '\n';
    return 0;
}