算法训练 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。

思路解析

这是一道动态规划题目，想到了思路其实特别简单。
首先我不好直接解决长度为L的K好数有几种可能性。但是可以通过求出长度为L-1的K好数的数目再此基础上往末尾再添一位符合题目条件的数即可。
例如4进制的3位K好数，末尾位是2的可能性m[3][2]=m[2][0]+m[2][2]。m[2][2]=m[1][0]+m[1][2]，由此可以总结出递推方程如下：

设表示K进制i位K好数其末尾位是j的所有可能性是m[i][j]
则：

代码

#include <iostream>
using namespace std;
#include<stdio.h>
#include<stdlib.h>
#include<cstring>
#include<math.h>
//动态规划
//用m[i][j]来表示长度为i的末尾为j的所有可能性数
//长度为0的末尾为j的m[0][j]=0;
//长度为0的末尾为j的m[1][j]=1(m[1][0]=0除外);
//体现不断往末尾加一位的思想，加的这一位不能是其前一位的相邻的数
//从m[i-1][j]到m[i][j]:m[i][j]=m[i-1][k](其中|k-j|!=1);
//最后结果sum=所有m[L][0...K-1]之和
long long m[101][101];
int main()
{
   memset(m,0,sizeof(m));
   int K,L;//K表示K进制，L表示L位数
   cin>>K>>L;
   //长度为0的末尾为j的m[0][j]=0;
   for(int j=0;j<=K-1;j++)
      m[0][j]=0;
   //长度为0的末尾为j的m[1][j]=1(m[1][0]=0除外);
   m[1][0]=0;
   for(int j=1;j<=K-1;j++)
       m[1][j]=1;
   //开始规划
   for(int i=1;i<=L;i++)//最后要求的长度为L
   for(int j=0;j<=K-1;j++){//每一位的取值可以是0-(K-1)
      for(int t=0;t<=K-1;t++)//往末尾加一位的可能取值也是0-(K-1)
        if(t-j!=-1&&t-j!=1){
             m[i][j]=m[i][j]+m[i-1][t];
             m[i][j]=m[i][j]%1000000007;//以防整数过大取余
        }
   }
   long long sum;//定义结果
   sum=0;
   //sum的值为数组m的第L行的从0-(K-1)所有元素之和
   for(int i=0;i<=K-1;i++){
     sum=sum+m[L][i];
     sum=sum%1000000007;//以防整数过大取余
   }
   cout<<sum;
   return 0;
}