Ural 1009 K-based Numbers(DP)

题目地址:Ural 1009

DP水题。。二维DP,第一维只用0和1来表示0和非零就可以。

代码如下:

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <stdlib.h>
#include <math.h>
#include <ctype.h>
#include <queue>
#include <map>
#include <set>
#include <algorithm>

using namespace std;
#define LL long long
LL dp[3][20];
int main()
{
    LL n, k, i;
    while(scanf("%lld%lld",&n,&k)!=EOF)
    {
        dp[0][1]=0;
        dp[1][1]=k-1;
        for(i=2;i<=n;i++)
        {
            dp[0][i]=dp[1][i-1];
            dp[1][i]=(dp[1][i-1]+dp[0][i-1])*(k-1);
        }
        printf("%lld\n",dp[0][n]+dp[1][n]);
    }
    return 0;
}


你可能感兴趣的:(编程,算法,dp,C语言,ACM)