首页 > 代码库 > 【Ural 1057】幂和的数量
【Ural 1057】幂和的数量
【题目描述】
写一个程序来计算区间[X,Y]内满足如下条件的整数个数:它恰好等于K个互不相等的B的整数幂之和。
举个例子。令X=15,Y=20,K=2,B=2。在这个例子中,区间[15,20]内有3个整数恰好等于两个互不相等的2的整数幂之和:
17=2^4+2^0
18=2^4+2^1
20=2^4+2^2
【输入格式】
输入文件的第一行有两个空格隔开的整数X,Y(1<=X<=Y<=2^31-1).
第二行有两个整数K,B(1<=K<=20,2<=B<=10).
【输出格式】
输出一行一个整数,即[X,Y]中恰好等于K个互不相等的B的整数幂之和的数的个数。
【分析】
数位类统计,上一张图:
用f[i][j]来表示高度为i的二叉树下进制位有j个为1的数的个数。
对于询问n,我们需要求出不超过n的最大B进制表示只含0、1的数:
找到n 的左起第一位非0、1 的数位,将它变为1,并将右面所有数位设为1。(因为大于1的肯定不可取,后面置为1使它最接近原来的数)
将得到的B进制表示视为二进制进行询问即可。
1 #include <cstdio> 2 #include <iostream> 3 #include <cstring> 4 int f[32][32]; 5 int x,y,k,b; 6 7 int work(int x,int k); 8 int change(int x); 9 int main() 10 { 11 int i,j; 12 //初始化 13 f[0][0]=1; 14 for(i=1;i<=31;i++) 15 { 16 f[i][0]=f[i-1][0]; 17 for(j=1;j<=i;j++) 18 f[i][j]=f[i-1][j]+f[i-1][j-1]; 19 } 20 scanf("%d%d%d%d",&x,&y,&k,&b); 21 printf("%d\n",work(change(y),k)-work(change(x-1),k)); 22 return 0; 23 } 24 int change(int x) 25 { 26 int p=1,tot=0; 27 while(x>=(long long)p*b) p*=b,++tot;//用来统计有tot个b进制位 28 int ans=0; 29 //b进制转2进制 30 while(p && x/p<=1) 31 { 32 ans+=x/p*(1<<tot); 33 tot--; 34 x%=p; 35 p/=b; 36 } 37 ans+=(1<<(tot+1))-1; 38 return ans; 39 } 40 //统计[0..x]内二进制表示含k个1的数的个数 41 int work(int x,int k) 42 { 43 //tot记录当前路径上已有的1的数量 44 int ans=0,tot=0; 45 for(int i=31;i;i--) 46 { 47 if(x&(1<<i)) 48 { 49 ++tot; 50 //跳出 51 if(tot>k)break; 52 x^=(1<<i); 53 } 54 if((1<<(i-1))<=x) 55 ans+=f[i-1][k-tot]; 56 } 57 if(x+tot==k)++ans; 58 return ans; 59 }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。