k好数

题意:如果一个自然数N的K进制表示中任意的相邻的两位都不是相邻的数字,那么我们就说这个数是K好数。求L位K进制数中K好数的数目。例如K = 4,L = 2的时候,所有K好数为11、13、20、22、30、31、33 共7个。由于这个数目很大,请你输出它对1000000007取模后的值。

输入包含两个正整数,K和L。

输出一个整数,表示答案对1000000007取模后的值。

思路:很明显这是一道DP的题,我们一层一层去DP,我们设dp[i][j]第i层如果为j的方法数  那么最后我们只需要把所有dp[k][i]加起来就可以

下面看代码:代码中有许多解释

#include
#include
#include
#define mod 1000000007
long long dp[105][105];
int main()

{
    int n,i,j,k,l;
    scanf("%d%d",&k,&l);
   // printf("%d %d",k,l);
    for(i = 0;i < k;i++)
    {
        if(i == 0)
        {
            dp[1][i] = 0;
            continue;
        }
        dp[1][i] = 1;
    }
    for(i = 2;i <= l;i++)   //计算第几层的
    {
        for(j = 0;j < k;j++)    //计算这层是哪个数
        {
            for(int x = 0;x < k;x++)  //计算哪个数是可以相加的
            {
                if(x != j - 1&&x != j + 1)
                {
                    dp[i][j] += dp[i - 1][x];
                    dp[i][j] = dp[i][j] % mod;
                }
            }
        }
    }
    long long sum = 0;  // 这个地方一定要用longlong
    for(i = 0;i < k;i++)
    {
        //printf("%d   *\n",dp[l][i]);
        sum = sum + dp[l][i];
        sum = sum % mod;
    }
    printf("%I64d\n",sum);
}



你可能感兴趣的:(算法)