演算法訓練 K好數(C/C++)AC碼
藍橋杯演算法訓練K好數AC碼
題目要求:
演算法訓練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。
接下來是程式碼,每一步的思路都在程式碼註釋中。
#include <stdio.h>//AC碼——Accept #define mod 1000000007 int main() { int dp[105][105]={0};//初始化dp陣列105——數量級在100以內 int k,l;//進位制 k 位數 l int i,j,x; scanf("%d%d",&k,&l); for(i=0;i<k;i++) dp[1][i]=1; //單獨考慮 L=1 位時候的情況 for(i=2;i<=l;i++) for(j=0;j<k;j++)//第i層0~k迴圈 for(x=0;x<k;x++)//第i-1層滿足條件的數 if(x!=j-1&&x!=j+1)//根據題意,本位的數字與前面的數字是不能相鄰的 { dp[i][j]+=dp[i-1][x];//累計符合方案數 dp[i][j]%=mod;//每次取模比最後取模好 } int sum=0; for(i=1;i<k;i++) { sum+=dp[l][i];//所有方案數加起來 sum%=mod;//最後取模 } printf("%d\n",sum); return 0; }