K-进制数-Java

题目描述：

考虑包含N位数字的K-进制数. 定义一个数有效, 即其K-进制表示不包含两连续的0.

例: 
1010230 是有效的7位数 
1000198 无效 
0001235 不是7位数, 而是4位数. 

给定两个数N和K, 要求计算包含N位数字的有效K-进制数的总数. 

假设2 <= K <= 10; 2 <= N; 4 <= N+K <= 18.

输入：

两个十进制整数N和K

输出：

十进制表示的结果

样例输入：

2
10

样例输出：

90

参考解答：

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int k = sc.nextInt();
		System.out.println(sum(n, k));
	}
	private static int sum(int n, int k) {
		int[] arr = new int[n];
		arr[0] = k - 1;
		arr[1] = ((k - 1) + 1) * (k - 1);
		for (int i = 2; i < n; i++) {
			arr[i] = (arr[i - 1] + arr[i - 2]) * (k - 1);
		}
		return arr[n - 1];
	}
}

思路参考：