URAL 1057 数位DP

题意 问[x,y]区间内，求B进制数，只由0和1组成，且有k个1，这样·的数有多少个。

思路 这题很有意思的一点是其实二进制和其它进制处理方式基本没有大的差别，只是其它进制当出现一位大于1时，就break。

f(i,j,k)表示i位长，第一位是否大于等于1（j=0或1），有k个1时，一共有多少个符合题意的数。（其实后来想了想j这一维可以不要）。

这题小碎点挺多的，卡了半天....比如注意超int的问题....

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

const int maxb = 11;
const int maxk = 21;
const int maxlen = 32;

typedef unsigned int ui;
const ui inf = ((ui)1<<(ui)31) - (ui)1;
ui f[maxlen][2][maxk];

int K,B;
ui n,m;

void yu()
{
	for(int j=0;j<2;j++)
		f[0][j][0] = 1;
	for(int i=1;i<maxlen;i++)
	{
		for(int j=0;j<2;j++)
		{
			for(int k=0;k<min(maxk,i+1);k++)
			{
				if(k > 0)
				{
					if(j==0)
						f[i][j][k] = f[i-1][1][k];
					else
						f[i][j][k] = f[i-1][1][k-1] + f[i][j-1][k];
				}
				else
				{
					if(j==0)
						f[i][j][k] = f[i-1][1][k];
					else
						f[i][j][k] = f[i][0][k];
				}
			}
				
		}
	}
	
}

int digit[maxlen];
int chaifen(ui x){
	int i = 0;
	while(x > 0)
	{
		i++;
		digit[i] = (int)(x%(ui)B);
		x/=(ui)B;
	}
	return i;
}

ui go(ui x)
{
	int len = chaifen(x);
	ui ans = 0;
	int now = 0;
	for(int i=len;i>=1;i--)
	{
		//这里的break!!注意！ 
		if(digit[i]>1)
		{
			ans += f[i][1][K-now];
			break;
		}
		else if(digit[i] == 1)
		{
			ans += f[i][0][K-now];
			now++;
		}
		if(now > K)
			break;
	}
	return ans;
}
int main()
{
	yu();
	while(scanf("%u%u%d%d",&n,&m,&K,&B)!=EOF)
	{
		printf("%u\n",go(m+1)-go(n));
	}
	return 0;
}