问题描述
如果一个自然数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;
int dp[105][105];
int main()
{
int k,l;//k进制数,l位
cin>>k>>l;
for(int j=0;j<k;j++)
{
dp[0][j]=1;
}
for(int i=1;i<l;i++)
for(int j=0;j<k;j++)
for(int t=0;t<k;t++)
{
if(t!=j-1&&t!=j+1)
dp[i][j]+=dp[i-1][t];
dp[i][j]%=1000000007;
}
int sum=0;
for(int j=1;j<k;j++)
{
sum+=dp[l-1][j];
sum%=1000000007;
}
cout<<sum<<endl;
}
对于动态规划,现在还基本不太会找规律,不太会推表达式
看了下面两篇博客,学会了些思路:
蓝桥杯,k好数,动态规划 - Marvin-wen - 博客园蓝桥杯之K好数问题_Walk the horizon-CSDN博客
首先以此样例为例来分析一下用DP做的过程:
L位K进制数,各位上数字的范围为[0,k-1],位数范围为[1,L]
以2位四进制数为例,列出如下表格:
0 | 1 | 2 | 3 | |
0 | 1 | 1 | 1 | 1 |
1 | 3 | 2 | 2 | 3 |
首先第一位0到3都可以取,因此都为1。下面分析第二位:
若第一位为0,第二位可以取0、2、3,有3种情况,dp[1][0]=dp[0][0]+dp[0][2]+dp[0] [3]=1+1+1=3;
若第一位为1,第二位可以取1、3,有2种情况,dp[1][1]=dp[0][0]+dp[0][3]=1+1=2;
若第一位为2,第二位可以取0、2,有2种情况,dp[1][2]=dp[0][0]+dp[0][2]=1+1=2;
若第一位为3,第二位可以取0、1、3,有3种情况,dp[1][3]=dp[0][0]+dp[0][1]+dp[0] [3]=1+1+1=3;
由于第一位不能为0,因此最后可取总数为dp[1][1]+dp[1][2]+dp[1][3],为7。