SPOJ-1182 Sorted bit squence 数位DP

这题虽然说是什么按位DP，其实尼玛不是组合数学么。不过硬是用模板的按位DP实现了，其实也就是记忆化搜索，本题恶心就在于有负数的存在，其实对付它就是把正数的第33位都变成1，用long long来处理，这样既保证了负数小于正数，又可以化成单一的区间了。在按位统计的时候记得当1出现在33位的时候不统计这个1。这题思路也就是先把[a, b]区间内含有一个1，两个1，三个1...的数的个数全部统计出来，一个for循环就能够找出该数字出现在有几个1的一段上，然后再二分查找答案，最后输出。

代码如下：

#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

typedef unsigned int Int;

Int a, b;

int bit[35], sum[35], dp[35][35], k; // 如何用记忆化去优化搜索
// sum[i]表示i个1的数有多少个
// dp[i][j] 代表剩余i位在没有约束的条件下产生j个1的组合数

long long x, y, ret;

void aly(int pos, int statu, int limit, int sign)
{
if (pos == -1) {
sum[statu] += sign;
return;
}
if (!limit && dp[pos][0] != -1) { // dp[i][0] - dp[i][32]是成套更新的
for (int i = 0; i <= 32; ++i) {
sum[i+statu] += sign * dp[pos][i];
}
return;
}
int s, end = limit ? bit[pos] : 1;
if (!limit) {
for (int i = 0; i <= 32; ++i) {
dp[pos][i] = sum[i]; // 先记录
}
}
for (int i = 0; i <= end; ++i) {
s = statu;
if (i == 1 && pos != 32) s = statu + 1;  // 只需要对状态更新进行判定，最高位（32位）为虚位，不需要计算
aly(pos-1, s, limit && i == end, sign);
}
if (!limit) {
for (int i = 0; i <= 32; ++i) {
dp[pos][i] = abs(sum[i+statu] - dp[pos][i]);    // 可能带来负值，计算出增加的值
}
}
}

void Cal(long long z, int sign)
{
int idx = -1;
while (z) {
bit[++idx] = z % 2;
z /= 2;
}
aly(idx, 0, 1, sign);
}

Int Ac(long long z, int loc, int Len)
{
int idx = -1;
memset(sum, 0, sizeof (sum));
Cal(z, 1), Cal(x-1, -1);
return sum[Len] >= loc;
}

long long bsearch(long long l, long long r, int loc, int Len) // 有Len个1的，第loc个数
{
long long mid, ret;
while (l <= r) {
mid = (l + r) >> 1;
if (Ac(mid, loc, Len)) {
ret = mid;
r = mid - 1;
}
else {
l = mid + 1;
}
}
return ret;
}

int main()
{
memset(dp, 0xff, sizeof (dp));
int T;
scanf("%d", &T);
while (T--) {
memset(sum, 0, sizeof (sum));
x = y = 0;
scanf("%u %u %u", &a, &b, &k);
x |= a, y |= b;
if (int (b) >= 0) {
y += 1LL << 32; // 给所有的正数加上一位的虚位
}
if (int (a) >= 0) {
x += 1LL << 32;
}
Cal(y, 1), Cal(x-1, -1);  // 已经统计出了区间内含有各个1的数字的数量
for (int i = 0; i <= 32; ++i) {
k -= sum[i];
if (k <= 0) {
k += sum[i]; // 去区间内寻找第k个位数为i的数字
ret = bsearch(x, y, k, i);
printf("%d\n", int (ret));
break;
}
}
}
return 0;
}