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

一开始看到这道题我由于没学过动态规划的思想，把这道题想的有点复杂，居然想着把那个指定进制和位数的数重大到小一个一个检查，毫无疑问超时了，代码如下：

#include<iostream>

using namespace std;

typedef unsigned long long ull;

void ff(int a[],int i,int k);
ull f(int k,int l);
int flag=1;

int main()
{
    ull sum;
    int k,l;
    cin>>k>>l;
    sum=f(k,l);
    cout<<sum;
    return 0;
}

ull f(int k,int l)
{
    ull sum=0;
    int a[100];
    int i,j,n;
    for(i=0;i<l;i++)
    {
        a[i]=k-1;//把值赋到最大
    }
    i=i-1;
    while(1)
    {
        ff(a,i,k);
        if(a[0]==0) break;
        flag=1;
        for(j=0;j+1<l;j++)//检查相邻数间，是否有相邻的数
        {
            if(a[j]+1==a[j+1]||a[j]-1==a[j+1])
            {
                flag=0;
                a[j+1]--;//把相邻俩个数的后者减一，可以稍微降低时间复杂度
                break;
            }
        }l;
        if(flag)
        {
            sum++;
            sum%=1000000007;
            a[i]--;
        }
    }
    return sum;
}

void ff(int a[],int i,int k)//实现数字的，规律减小，检查数字的合理性
{
    if(a[0]==0);
    else if(a[i]<=-1)
    {
        a[i]=k-1;
        a[i-1]--;
        ff(a,i-1,k);
    }
    else if(flag==0&&i>0)//如果有相邻数减一的情况，检查是否减到了-1；
    {
        if(a[i]==-1)
        {
            a[i]=k-1;
            a[i-1]--;
        }
        ff(a,i-1,k);
    }
}

上面是错误示范，我只关注整体，程序的时间复杂度非常大；

下面看看正确的做法

#include<iostream>

using namespace std;

int main()
{
    unsigned long int dp[200][200];//在设置数组长度这我其实有点迷糊，因为k和l都是小于100的，我想设成101就够了但是其结果不对，就感觉很懵，后来改大了长度就对了，如果有大佬能解释一下，我非常感激
    int k,l;
    cin>>k>>l;
    int i,j,m;
    unsigned long int res=0;
    dp[0][0]=0;//不能以0开头
    for(i=1;i<k;i++)//i+1就表示实际位数
    {
        dp[0][i]=1;//1位时是问题分解到最小的结果，从1开始给每个赋值为1，表示他是一个符合条件的数
    }
    for(i=1;i<l;i++)//从两位开始往后，新加数字放到末尾
    {
        for(j=0;j<k;j++)//j为所求指定位数的末尾数字
        {
            for(m=0;m<k;m++)//m为指定位数的上一位的数字
            {
                if(m+1==j||m-1==j) continue;//想邻位数的数字相邻就跳过
                dp[i][j]=(dp[i][j]+dp[i-1][m])%1000000007;//dp数组中装的是符合条件的k好值的个数
            }
        }
    }
    for(i=0;i<k;i++)
    {
        res=(res+dp[l-1][i])%1000000007;//计算最终结果
    }
    cout<<res<<endl;
    return 0;
}

是的代码非常少，而且时间复杂度降低了很多，他是用到动态规划的思想，把一个大问题看作一个个小问题，逐一解决。
在这道题中，我以4进制3位举例：
其结果可以想作是3位的0，1，2，3结尾的个数和；
0结尾的，又可以看作是2位以0，2，3结尾的数字的个数和，二位的符合条件数字的有20、30、22、13、33。五个数，也就是3位以0结尾的有五个k好数。
1结尾的，可以看作2位以1，3结尾的数字的个数和，二位的符合条件的数字有11，31，13，33。4个数，也就是3位以1结尾的有四个k好数。
再看三位以2结尾的的，可以看作是2位以0、2结尾的数字的个数和，符合条件的数字有30、20、22，即三位以2结尾的k好树有三个。
最后只差三位以3结尾的数字，即二位以0，1，3结尾的数字个数和，符合条件的有20、30、11、31、13、33，六个数

所以最终的结果为3为以0、1、2、3的数字的总个数
为5+4+3+6=18
在代码中表示为res=dp[2][0]+dp[2][1]+dp[2][2]+dp[2][3]