算法训练 K好数

【问题】

问题描述

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

输入格式

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

输出格式

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

样例输入

4 2

样例输出

7

数据规模与约定

对于30%的数据，KL <= 106；
对于50%的数据，K <= 16， L <= 10；
对于100%的数据，1 <= K,L <= 100。

【问题分析】
1、首先，理解题意是本题的一个难点。K进制数的取值范围为[0，k-1],而L位K进制数意思是：这个数位数有L位，并且每一位数都是K进制数，即每一位都属于[0,k-1].
2、对于取模运算的必要，我们知道，K、L的取值范围为1<=K,L<=100,当K=100,L=100时，这样的数的总数有100的100次方个，显然超出了int、long long 的范围，故进行取余运算。
3、这题为动态规划题，解题思路为dp[i][j]=dp[i][j]+dp[i-1][k];这里的i代表所在位数(i=0表示个位，i=1表示十位，以此类推)，j表示取K进制下的一个数，K的意思与j相同。整个式子表示：当前位置的个数=当前位置的个数+前一个位置的个数总数。
例如，以2位4进制的4好数为例
dp[1][0]=dp[1][0]+dp[0][k];
//表示十位数字为0时的4好数个数=表示十位数字为0时的4好数个数+个位数字为K的数的总数，k属于[0,k-1];
dp[1][1]=dp[1][1]+dp[0][k];//理解同上
dp[1][2]=dp[1][2]+dp[0][k];
dp[1][3]=dp[1][3]+dp[0][k];
4、对于1位1进制数，蓝桥系统给的测试结果为0个，这里我也有点疑问，可能不把0算在内吧。。。

【代码】

#include <iostream>
#include <cmath>
using namespace std;
int dp[101][101];
 #define MZ 1000000007
 int Count(int l,int k,int sum)
 {
 
     int i,j,m;
       for(i=1;i<k;i++)//注意从1开始，0不算在内
     {
         dp[0][i]=1;
     }
     for(i=1;i<l;i++)
        for(j=0;j<k;j++)
        for(m=0;m<k;m++)
     {
         if(abs(j-m)!=1)//K好数任意两个数不相邻
         {
           dp[i][j]=dp[i][j]+dp[i-1][m];
           dp[i][j]%=MZ;
         }
     }
     for(i=0;i<k;i++)
     {
         sum+=dp[l-1][i];
         sum%=MZ;
     }
     return sum;
 }
int main()
 {
    int sum=0;
     int k,l,i,j,m;
     cin>>k>>l;
     if(k==1&&l==1)
        cout<<0<<endl;//1位1进制数结果为0
     else if(k>1&&l==1)
        cout<<k-1<<endl;//1位K进制数结果为k-1,注意蓝桥杯系统结果为K也对，但上面必须为为0，迷之操作。。。
     else if(l>1)
        cout<<Count(l,k,sum)<<endl;
 }