ural 1009 K-based Numbers

类型：递推 

题目：求n位的合法（无两个连续的0出现）k进制个数

状态：f(i)表示合法的i位k进制数（首位不为0）的个数 

容易知道第i位有k - 1种选择，考虑第i - 1位，如果第i - 1位为0，情况转化为f(i - 2)，如果不为0，情况转化为f(i - 1)

故有：f(i) = (k - 1)(f(i - 1) + f(i - 2))

// ural 1009 K-based Numbers
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

#define FOR(i,a,b) for(i = (a); i < (b); ++i)
#define FORE(i,a,b) for(i = (a); i <= (b); ++i)

int n, k;
int f[30];

int main()
{
    int i;

    while(cin>>n>>k) {
        f[1] = k - 1;
        f[2] = k * (k - 1);
        FORE(i, 3, n)
            f[i] = (k - 1) * (f[i - 1] + f[i - 2]);
        cout<<f[n]<<endl;
    }
    return 0;
}