上海高校金马五校赛 F 1 + 2 = 3?【规律+二分+数位DP】

https://www.nowcoder.com/acm/contest/91/F

链接：https://www.nowcoder.com/acm/contest/91/F
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld
题目描述
小Y在研究数字的时候，发现了一个神奇的等式方程，他屈指算了一下有很多正整数x满足这个等式，比如1和2，现在问题来了，他想知道从小到大第N个满足这个等式的正整数，请你用程序帮他计算一下。

(表示按位异或运算)

输入描述:
第一行是一个正整数，表示查询次数。

接着有T行，每行有一个正整数，表示小Y的查询。

输出描述:
对于每一个查询N，输出第N个满足题中等式的正整数，并换行。
示例1
输入
4
1
2
3
10
输出
1
2
4
18

通过规律可以发现，凡是有相邻数字1的就不可以实现上述等式。因此这个题可以用二分答案，并且通过数位DP来solve了。

许久没做过数位DP了，非常的生！

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
#include<queue>
#include<vector>
using namespace std;
#define ll long long int
#define INF 0x3f3f3f3f
#define Irish_Moonshine main
const int maxn = 1e5 + 2333;
const int p = 1e9 + 7;
ll dp[110][2];
ll a[110];
ll n;
ll dfs(int len, int sta, bool lim);
ll solve(ll x) {
    memset(dp, -1, sizeof dp);
    int cnt = 0;
    while (x > 0) {
        a[cnt++] = x % 2; x /= 2;
    }
    return dfs(cnt - 1, 0, 1);
}
ll dfs(int len, int sta, bool lim) {
    if (len < 0) return 1;//递归边界，既然是按位枚举，最低位是0，那么pos==-1说明这个数我枚举完了  
    if (dp[len][sta] != -1 && lim == 0) {
        return dp[len][sta];
    }
    int up = lim ? a[len] : 1;//根据limit判断枚举的上界up
    ll ans = 0;
    for (int i = 0; i <= up; i++) {//枚举，然后把不同情况的个数加到ans就可以了  
        if (sta&&i == 1) continue;
        ans += dfs(len - 1, i == 1, lim&&i == up);
    }
    return lim ? ans : dp[len][sta] = ans;//刚开始最高位都是有限制并且有前导零的，显然比最高位还要高的一位视为0嘛  
}
int Irish_Moonshine()
{
    int T;
    scanf("%d", &T);
    while (T--) {
        scanf("%lld", &n);
        ll l = 1, r = 1e18;
        ll mid, ans;
        while (l <= r) {
            mid = (l + r) / 2;
            ll t = solve(mid) - 1;
            if (t < n) {
                l = mid + 1;
            }
            else {
                ans = mid;
                r = mid - 1;
            }
        }
        printf("%lld\n", ans);
    }
    return 0;
}
//_CRT_SECURE_NO_WARNINGS