蓝桥杯 算法训练 K好数 AC码
题目要求:
接下来是代码,每一步的思路都在代码注释中。
#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;
}