试题 算法训练 K好数

试题 算法训练 K好数
资源限制
时间限制：1.0s 内存限制：256.0MB

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

得过一次90，因为没有判断k==1的情况。
因为做的算法题少吧，dp做法是头一次见，翻了好多资料知道最重要是找出状态转移方程，再多练习应该就能上手自己做了，这个是参考了其他人思路写出来的。

#include<iostream>
#include<cmath>
using namespace std;
int main(){
 	int k,l;
 	cin>>k>>l;
 	long long dp[101][100]={0}; 
 	long long count=0;
 	//k进制l位
 	for(int j=0;j<k;j++){
  		dp[1][j]=1;
 	}
 	if(l>=2){
  		for(int i=2;i<l+1;i++){
   			for(int j=0;j<k;j++){
    				for(int z=0;z<k;z++){
     					if(z!=j+1 && z!=j-1){
      						dp[i][j]+=dp[i-1][z];
      						dp[i][j]%=1000000007; 
     					}
      
    				}
   			}
  		} 
 	}
 	if(k==1)
  		cout<<"0";
	else{
  		if(l==1)
   			cout<<k;
  		else{
   			for(int j=1;j<k;j++){
    				count+=dp[l][j];
   			}
   			cout<<count%1000000007;
  		}
  	}
  	return 0;
 }