Codeforces#458 C.Travelling Salesman and Special Numbers

题目：
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;
}