K-based Numbers(Version1&&Version2)

题面很简单，就是问你k进制下的n位数有多少是有效地(没有两个0相邻且最高位不为0)

解法:我们现在对第i(i!=1&&i!=2&&i!=n)位进行讨论，如果第i位是0的话，那么第i-1位则不能是零,则第i位为0情况下其有效数字个数为第i-1位不为0的有效数字个数和,如果第i位不为0,则第i位可以是0-k-1间的所有数，则第i位不为0时其有效数字个数为i-1位所有情况下的有效数字总和，结合上面讨论得到在i-1=0的情况下的有效数字个数，推导得出第i位不为0的有效数字总和为第i-1位不为0+第i-2位不为0情况小的总和再乘以(k-1),所以可以等到状态转移方程为dp[i]=(dp[i-1]+dp[i-2])(k-1)且易知dp[1]=k-1,dp[2]=k(k-1);

AC代码

#include<iostream>
#include<cstring>
#include<cstdio>

using namespace std;

typedef long long ll;

ll dp[5000];

int main() {
	int n,k;
	while (scanf("%d %d", &n, &k) != EOF) {
		dp[1] = (ll)k - 1;
		dp[2] = (ll)k * (k - 1);
		for (int i = 3; i <= n; i++) {
			dp[i] = (ll)(dp[i - 1] + dp[i - 2]) * (k - 1);
		}
		printf("%d\n", dp[n]);
	}
}

然后是第二个版本，这个版本和第一个版本相比n,k范围增大，所以要用到高精，然后偷懒的我把上个代码用java大数类又给实现了一遍，最后AC

AC代码

import java.math.BigInteger;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        Scanner scanner = new Scanner(System.in);

        BigInteger n ;

        n=scanner.nextBigInteger();

        BigInteger k;

        k = scanner.nextBigInteger();

        BigInteger dp[]= new BigInteger[2000];

        dp[1]=k.subtract(BigInteger.ONE);

        dp[2]=k.multiply(dp[1]);

        for(int i=3;i<=n.intValue();i++){
            dp[i]=(dp[i-1].add(dp[i-2])).multiply(dp[1]);
        }

        System.out.println(dp[n.intValue()]);

    }
}