DP-K 好数

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。

思路：

其一：穷举，太麻烦。。

其二：动态规划

 记a[i][j]表示第i位是j的时候的数量；

显然有 a[i][j]=∑(a[i-1][p]),其中 0<=p>K 并且 p与j不相邻

实现：

#include <iostream>
#include <string.h>
using namespace std;
#define MAX_N 10010
#define M 1000000007
int n,m;
int f[MAX_N][MAX_N];
int main(){
    while(cin>>n>>m){//n进制 m位
        for(int i=0;i<n;i++){
            f[1][i]=1;
        }
        f[1][0]=0;
        int m1;
        for(int i=2;i<=m;i++){
            for(int j=0;j<n;j++){
                for(int k=0;k<n;k++){
                    if(j==0 && k==1)
                        continue;
                    else if(j==n-1 && k==n-2)
                        continue;
                    else if(k==j+1 || k==j-1)
                        continue;
                    else
                        f[i][j]+=f[i-1][k];
		    f[i][j]%=M;
                }
                f[i][j]%=M;
            }
        }
/*
        for(int i=1;i<=m;i++){
            for(int j=0;j<n;j++)
                cout<<f[i][j]<<" ";
            cout<<endl;
        }
*/
        int sum=0;
        for(int i=0;i<n;i++){
            sum+=f[m][i];
            sum%=M;
        }
        cout<<sum<<endl;
    }
return 0;}

上述如三重循环，可以把求和的部分优化，改为两重循环，总和求出来，把相邻的减去，但是要注意可能因为减法出现的负数：

#include <iostream>
#include <string.h>
using namespace std;
#define MAX_N 10010
#define M 1000000007
int n,m;
long long f[MAX_N][MAX_N];
int main(){
    while(cin>>n>>m){//n进制 m位
        memset(f,0,sizeof(f));
        for(int i=0;i<n;i++){
            f[1][i]=1;
        }
        f[1][0]=0;
        int s=n-1;
        for(int i=2;i<=m;i++){
            int sum=0;
            for(int j=0;j<n;j++){
                if(j==0)f[i][j]=s+M-f[i-1][1];
                else if(j==n-1) f[i][j]=s+M-f[i-1][n-2];
                else
                    f[i][j]=s+M-f[i-1][j-1]-f[i-1][j+1];
                f[i][j]+=M;
                f[i][j]%=M;
                sum+=f[i][j];
                sum%=M;
            }
            s=sum;
        }
        cout<<s<<endl;
    }
return 0;}