http://acm.hdu.edu.cn/showproblem.php?pid=5106
给定n和r,要求算出[0,r)之间所有n-onebit数的和,n-onebit数是所有数位中1的个数。
对于一个n-bit数,可以根据与R最高不同位的位置分成几类。比如R=100100010,可以分成0xxxxxxxx,1000xxxxx,10010000x三类。x处可任取0或者1。x的个数为n,x中1的个数为k。和分两部分,高于n位的有C(n,k)乘以n位之前的数字,低于n位的部分有C(n-1,k-1)*(2^n−1) ( 对于k个1中的某一个1的位置,这个位置上有1在所有情况中占有了C(n-1,k-1)种情况,那么就相当于11...11(n个位置每个位置上都可能有1)*C(n-1,k-1),即C(n-1,k-1)*(2^n−1) )
三次相乘一定要在第二次中间加一个%mod,血的教训!
生成组合数的公式
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#define RD(x) scanf("%d",&x)
#define RD2(x,y) scanf("%d%d",&x,&y)
#define RD3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define clr0(x) memset(x,0,sizeof(x))
#define clr1(x) memset(x,-1,sizeof(x))
using namespace std;
typedef long long LL;
const int maxn = 1005;
const double eps = 1e-3;
const int inf = 0x7fffffff;
char p[maxn];
const LL mod = 1000000007;
LL c[maxn][maxn];
LL q[maxn];
void init()
{
for(int i = 0 ; i < maxn ; ++i){
c[i][0] = 1;
}
for(int i = 1 ; i < maxn ; i++){
for(int j = 1 ; j <= i ; ++j){
c[i][j] = (c[i-1][j-1] + c[i-1][j])%mod;
}
}
// c[0][0] = 1;
// for(int i = 1;i < maxn;++i)
// for(int j = 0;j < i;++j){
// c[i][j] = (j == 0)?c[i-1][j] : c[i-1][j-1] + c[i-1][j];
// c[i][j] %= mod;
// }
q[0] = 1;
for(int i = 1;i < maxn;++i){
q[i] = (q[i - 1]*2)%mod;
}
}
int main()
{
init();
int n,k;
while(~scanf("%d%s",&n,p)){
int m = strlen(p);
int k = 0;
LL ans = 0,tmp = 0;
for(int st = 0;st < m;++st){
if(p[st] == '1'){
ans += ((m-st-2>=0&&n-k-1>=0?c[m - st - 2][n - k - 1]:0) * (q[m - st - 1] - 1))%mod;
ans += (tmp * q[m - st]%mod * (m-st-1>=0&&n-k>=0?c[m - st - 1][n - k]:0))%mod;
ans %= mod;
}
tmp = (tmp*2 + p[st] - '0')%mod;
k += p[st] - '0';
}
if(ans < 0) ans += mod;
printf("%I64d\n",ans);
}
return 0;
}