URAL 1009 K-based Numbers

URAL_1009

    其实本质只有两类数，0和非0，所以可以用f[i][0]、f[i][1]表示递推到第i位时为0以及非0的情况种数进行dp。

#include<stdio.h>
#include<string.h>
#define MAXD 20
int N, K;
long long f[MAXD][2];
void solve()
{
    int i, j, k;
    f[1][1] = K - 1, f[1][0] = 0;
    for(i = 2; i <= N; i ++)
        f[i][0] = f[i - 1][1], f[i][1] = (K - 1) * (f[i - 1][0] + f[i - 1][1]);
    printf("%lld\n", f[N][0] + f[N][1]);
}
int main()
{
    while(scanf("%d%d", &N, &K) == 2)
    {
        solve();
    }
    return 0;
}