题目链接:http://acm.timus.ru/problem.aspx?space=1&num=1057
在《浅谈数位统计类问题》中的例1有详细题解,这里对题解中一些不怎么容易理解的地方加一些注释。
对于二进制的情况:
这课假想的完全二叉树自上而下对应二进制数字的从高位到低位。
“右转时遇到的”是指从根节点进入右子树时序统计根节点的左子树上相应的路径的个数。
函数 int calc(int x, int k)中,第 i 位是跟根节点(也是当前节点),第 i-1 位是孩子节点。如果第 i 位为1,那么路径中已经遇到的1的个数tot加1,同时抹除掉这个1,再判断第 i-1 位是否为1:若为1,那么从第 i 位到第 i-1 位是右转而来,统计左子树中路径中1的个数为k-tot有多少;若第 i 位为0,那么从第 i 位到第 i-1 位是左转,不必统计。最后再加上x本身,ans加1.
对于非二进制的情况:
从高位到低位查找,碰到第一个非0、1的数字时,将该位数字及其右边所有的数字全部变成1,然后按照二进制的方法来处理。这样为什么会正确?
如果按照二进制假想完全二叉树的方法来假想一科完全B叉树,那么这棵树包含了完全二叉树中所有的点,如果把所有以非0、1节点为根的子树删除掉,那么剩下的就是一棵与二进制中一样的完全二叉树,并且在这棵树中包含了所有合法的路径,只要在这棵树中统计不超过n的合法路径即可。而将n经过上面的处理之后得到的数就是在完全B叉树最靠近n的一条仅含有0、1的路径,这样就不会有遗漏了。
#include<cstdio>
using namespace std;
int f[35][35]; //f[i][j],高度为i并且1的个数为j的路径的个数
void init()
{
f[0][0] = 1;
for (int i=1; i<=31; i++)
{
f[i][0] = f[i-1][0];
for (int j=1; j<=i; j++)
f[i][j] = f[i-1][j] + f[i-1][j-1];
}
}
int calc( int x, int k)
{
int ans = 0, tot = 0;
for (int i=31; i>0; i--)
{
if (x & (1<<i))
{
tot++;
if (tot > k) break;
x ^= (1<<i);
}
if (1<<(i-1) <= x)
ans += f[i-1][k-tot];
}
if (tot + x == k) ans++;
return ans;
}
int ChengeTwo(int x, int b)
{
int a[35] = {0};
int ans = 0;
int i = 0;
while (x > 0)
{
a[++i] = x % b;
x /= b;
}
int j = i;
while (a[i]<=1)
{
i--;
}
while (i >= 1)
{
a[i] = 1;
i--;
}
while (j >= 1)
{
ans = ans * 2 + a[j];
j--;
}
return ans;
}
int main()
{
int x, y, k, b;
scanf("%d%d%d%d",&x, &y, &k, &b);
init();
if (b == 2)
printf("%d\n",calc(y,k)-calc(x-1,k));
else
{
x = ChengeTwo(x, b);
y = ChengeTwo(y, b);
printf("%d\n",calc(y,k)-calc(x-1,k));
}
return 0;
}