算法训练 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。

 题目大意

 输入k和l 代表k进制l位数

比如4进制2位数

那么就有 0 1 2 3 / 10 11 12 13 / 20 21 22 23/ 30 31 32 33这几种情况，满足条件的数有11、13、20、22、30、31、33 共7个

题目就是让我们求k进制l位数中的k好数

思路：dp

 首先想到的是全排列，当然这是一种暴力，看到如此大的数据量肯定超时，否决。

其次想到每次做乘法，大致思路是第一位有几种可能，然后在第一位的可能下，第二位又有几种可能

一定要考虑到前一位不同的情况，现在这一位计算出来又有不同的情况。

代码一是此解法（有点枚举的意思，最多想法好想但是不太好写，详解在后边）

然后我们用dp数组来做

代码二是此解法

dp[i][j]代表 以j结尾的长为i的数量，然后dp[i][j]=

然后我们一定要注意初始值

这个地方特别注意

void init(int k,int l)
{
    for(int i=0;i<k;i++) dp[2][i]=k-3;
    dp[2][0]=k-2;dp[2][1]=k-2;dp[2][k-1]=k-2; 
}

两位长度 以0结尾和以1结尾的都是k-2 我就是出错在这个地方

代码

代码一
 

#include<bits/stdc++.h>
using namespace std;
#define mmax 1000000007
int dp[110],tp[100];
long long solve(int k,int l)
{		 
	if(k<=2 || l<=1) return 0;
	for(int i=3;i<=l;i++)  //第几位 
	{
		for(int j=0;j<k;j++) //这一位是什么 
		{
			//int tmp=0;
			tp[j]=0;
			for(int x=0;x<k;x++)//前一位是什么 
			{
				if(abs(x-j)!=1)  tp[j]=(tp[j]+dp[x])%mmax; 
			}
			//tp[j]=tmp;
	
		}
		for(int q=0;q<k;q++) dp[q]=tp[q];
	}
	int ans=0;
	for(int i=0;i<k;i++) ans=(ans+dp[i])%mmax;
	return ans;
} 
void init(int k)
{
	for(int i=0;i<k;i++) dp[i]=k-3;
	dp[0]=k-2;dp[1]=k-2;dp[k-1]=k-2;
}
int main()
{
	int k,l;
	cin>>k>>l;
	init(k);
	cout<<solve(k,l); 
	return 0;
}

在 七进制 长度为5的情况下解释

首先解释dp数组 dp[i]代表当前长度下，最后一位是i的情况下的数量

思路是我先提前init初始化两位数字时候的结果，这个很容易知道，存在dp数组中

然后从第三位开始，我们首先枚举这一位数j,然后再枚举前一位数x，如果这两个数不相邻就让tp[j]累加前一个数的dp，

也就是说比如我们现在到了第四位，第四位枚举到是5，那么tp[5]应该等于所有与5不相邻的dp[x]之和（前一位是0 1 2 3 5 6都满足，那么dp[j]就应该是这些累加之和，但是我们不能现在就更改dp，因为还需要继续枚举当前位数，还需要用到前一位的dp，这就是为什么使用了tp数组）

当当前位数都枚举完成之后,dp数组就可以更新了，为=下一位做准备

 

最后结果是dp数组之和 因为dp数组 dp[i]代表最后一位是i的情况下的数量

 

 代码二

#include<bits/stdc++.h>
using namespace std;
#define mmax 1000000007
int dp[110][110];
void init(int k,int l)
{
	for(int i=0;i<k;i++) dp[2][i]=k-3;
	dp[2][0]=k-2;dp[2][1]=k-2;dp[2][k-1]=k-2; 
	
}
long long solve(int k,int l)
{		
	for(int i=3;i<=l;i++)
	{
		for(int j=0;j<k;j++)
		{
			int tp=0;
			for(int x=0;x<k;x++)
			{
				if(abs(x-j)!=1)	tp=(tp+dp[i-1][x])%mmax;
			}
			dp[i][j]=tp;
		}
	}
	int ans=0;
	for(int i=0;i<k;i++)
		ans=(ans+dp[l][i])%mmax;	
	return ans;
} 
int main()
{
	int k,l;
	cin>>k>>l;
	init(k,l);
	cout<<solve(k,l);
	
	return 0;
}

代码二