K好数（动态分配）——C++

问题描述

如果一个自然数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。

代码如下：

#include<bits/stdc++.h>
using namespace std;
int main()
{
 int K,L;
 int i,j,x,count=0;
 scanf("%d %d",&K,&L);
 int arr[L+1][K] ;    //[L+1]表示第几位数字(从1开始),[K]表示以什么数字结尾 
 memset(arr,0,sizeof(arr));
 for(i=0;i<K;i++)    //第一位只有一种可能 
  arr[1][i]=1;
 for(i=2;i<=L;i++)    //从第二位开始对数组赋值 
  for(j=0;j<K;j++)
   for(x=0;x<K;x++)  //从0开始检查 
   {
    if(x!=j-1&&x!=j+1) //如果x符合条件，将上一位以x结尾的数量增入 
     {
      arr[i][j]+=arr[i-1][x];
      arr[i][j]%=1000000007;
     }
   }
 for(i=1;i<K;i++)
  {
   count+=arr[L][i];
   count%=1000000007;
  } 
 printf("%d\n",count); 
 return 0;
} 

运行结果：

此文思路均源于：温酒煮清风 感谢！