K好数

来源:互联网 发布:鸣人只会搓丸子知乎 编辑:程序博客网 时间:2024/04/29 17:58
问题描述

如果一个自然数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>#include <cstring>#include <cstdio>#define N 110using namespace std;int dp[N][N];//i,共有i位,首位为jint fun(int n){    if (n < 0)        return 0 - n;    else         return n;}int main(){    int k, l;    scanf("%d%d", &k, &l);    if(k == 1 && l == 1)    {        printf("0\n");        return 0;    }    if(l == 1)    {        printf("%d\n", k);        return 0;    }    for (int i = 0; i < k; i++)        dp[1][i] = 1;    for (int i = 2; i <= l; i++)    {        for (int j = 0; j < k; j++)        {            for (int t = 0; t < k; t++)            {                if (fun(j - t) != 1)                    dp[i][j] = (dp[i][j] + dp[i - 1][t]) % 1000000007;            }        }    }    int sum = 0;    for (int i = 1; i < k; i++)        sum = (sum + dp[l][i]) % 1000000007;    printf("%d\n", sum);    return 0;}


0 0
原创粉丝点击