题目:
思路:dp思想
比如你要求k=4,l=3四进制三位数的话,分为三种情况3XX,2XX,1XX,以3XX(3开头)为例,要想知道3XX的个数,只要知道后两位XX的情况的个数即可。后两位可能是0X,1X,3X。然后要知道0X,只要知道X的个数即可,X可以是0,2,3共三个。同理,1X中的X有1,3两个,3X中的X有0,1,3三个。所以3XX的个数=0X+1X+3X=3+2+3=8.
此题中,用i表示长度(位数),j表示开头的数字,dp[i][j]表示这样的数的个数。m表示可取的数,因为k表示进制,所以m可取范围为[0…K-1]。k为4表示四进制,那么m范围为[0…3]。j表示开头可取的数字,所以j为m范围中去掉0,即[1…k-1]。
import java.util.Scanner;
//用i表示长度(位数),j表示开头的数。dp[i][j]表示个数。
public class K好数 {
public static void main(String[] args) {
int[][] dp=new int [105][105];
int mod=1000000007;
Scanner sc=new Scanner(System.in);
int k=sc.nextInt();
int l=sc.nextInt();
//dp[1][i]表示长度为1的数,故数量为1
for(int i=0;i<k;i++)
dp[1][i]=1;
for(int i=2;i<=l;i++) {//长度为i到l
for (int j = 0; j < k; j++) {//表示以j开头的个数,这里包括以0开头(0X,00)
for (int m = 0; m < k; m++) {//枚举以j开头的所有情况
if(m!=j+1&&m!=j-1) {
dp[i][j]+=dp[i-1][m];
dp[i][j]%=mod;
}
}
}
}
int ans=0;
//从1开始,因为开头(j)不能是0
for(int i=1;i<k;i++) {
ans+=dp[l][i];
ans%=mod;
}
System.out.println(ans);
}
}