如果一個自然數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。
k好數思路:要得到所有的L位K進制數的k好數,我們需要先得到L-1位K進制數的k好數,再將其中第L位與第L-1位相鄰的數去掉即可。(子問題)
如果理解了上面那句話,那么我們只需要從1位開始遞推到L位的K進制數即可。
定義一個二維數組dp[i][j],其中 i表示位數 j表示第一位數
即求出1位K進制數的k好數個數,dp[1][j],求出2位K進制數的k好數個數 dp[2][j],如此遞推下去直到L位,
再將保存L位開頭從1到K-1的K進制的k好數相加
AC代碼片段如下:
結合代碼更好理解一點
#include<iostream>#include<algorithm>#include<memory.h>using namespace std;#define mod 1000000007 typedef long long ll;ll dp[110][110]; //dp[i][j] 表示 i 位數 且 j為第一位的k好數個數int k; //k進制 int l; //l位數 int main(){ ll sum=0; //總計 cin>>k>>l; memset(dp,0,sizeof(dp)); for(int j=0;j<k;j++){ dp[1][j] =1; //1位數的K好數 } for(int i=2;i<=l;i++){ //從2位數到l位數,動態規劃 for(int j=0;j<k;j++){ //第一位數為j for(int f=0;f<k;f++){ //第二位數為f if(f+1!=j && f-1!=j){ //不是相鄰的數 dp[i][j] += dp[i-1][f]; dp[i][j] %= mod; } } } } for(int j=1;j<k;j++){ // 去掉第一位為0的k好數 sum += dp[l][j] ; sum %= mod; } cout<<sum; return 0;}
新聞熱點
疑難解答