题目:
http://codeforces.com/contest/914/problem/C
分析:
一次操作后,必定变为1000以内的数;
则先暴力处理出1000以内的合法数;
然后递推出组合数;
就能算啦;
注意k=1的时候,答案是 长度-1
代码:
#include <bits/stdc++.h>
using namespace std;
const int Tmax=1005;
const long long int MOD=1e9+7;
bool ok[Tmax];
char n[Tmax];
int k,len;
long long C[Tmax][Tmax],ans;
int go(int x)
{
int cnt=0;
while(x>0)
{
cnt+=(x&1);
x>>=1;
}
return cnt;
}
void init()
{
int i,j,tmp;
bool yes;
if(k==1) ok[1]=true;
for(i=2;i<=1000;i++)
{
tmp=i;
yes=true;
for(j=1;j<=k-1;j++)
{
tmp=go(tmp);
if(tmp<=1&&j!=k-1)
{
yes=false;
break;
}
}
if(yes==true&&tmp==1) ok[i]=true;
}
C[0][0]=C[1][0]=C[1][1]=1;
for(i=2;i<=1000;i++)
{
C[i][0]=C[i][i]=1;
for(j=1;j<i;j++)
C[i][j]=(C[i-1][j]+C[i-1][j-1])%MOD;
}
return;
}
void work()
{
int i,j;
len=strlen(n);
for(i=1;i<len;i++)
{
for(j=1;j<=i;j++)
if(ok[j]==true)
ans=(ans+C[i-1][j-1])%MOD;
}
int p=1,one=1;
while(p<len)
{
while(n[p]=='0'&&p<len) p++;
for(i=0;i<=len-p-1;i++)
{
if(ok[i+one]==true)
ans=(ans+C[len-p-1][i])%MOD;
}
one++;
p++;
}
int sum=0;
for(i=0;i<len;i++)
sum+=n[i]-'0';
if(ok[sum]==true) ans=(ans+1)%MOD;
return;
}
void run()
{
int i,j;
cout<<endl;
for(i=0;i<=10;i++)
{
for(j=0;j<=i;j++)
printf("%d ",C[i][j]);
cout<<endl;
}
return;
}
int main()
{
scanf("%s",n);
scanf("%d",&k);
if(k==0)
{
printf("1");
return 0;
}
else if(k>5)
{
printf("0");
return 0;
}
init();
work();
if(k==1) printf("%I64d",(ans-1+MOD)%MOD);
else printf("%I64d",ans);
//run();
return 0;
}