1. 程式人生 > >K好數(DP)

K好數(DP)

cpp script java 數據 我們 sin 數字 class 一個

問題描寫敘述
假設一個自然數N的K進制表示中隨意的相鄰的兩位都不是相鄰的數字,那麽我們就說這個數是K好數。

求L位K進制數中K好數的數目。

比如K = 4,L = 2的時候。全部K好數為11、13、20、22、30、31、33 共7個。因為這個數目非常大。請你輸出它對1000000007取模後的值。

輸入格式
輸入包括兩個正整數。K和L。

輸出格式
輸出一個整數,表示答案對1000000007取模後的值。
例子輸入
4 2
例子輸出
7
數據規模與約定
對於30%的數據,KL <= 106;

對於50%的數據,K <= 16, L <= 10;

對於100%的數據,1 <= K,L <= 100。


代碼實現

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#define MAX 110

using namespace std;

const int a=1000000007;
long long use[MAX];
int luse[MAX];
long long sum;
int K,L;
void refreash();
int main()
{
    memset( use,0,sizeof(use) );
    fill( luse,luse + MAX,1
); sum = 0; cin>>K>>L; for( int t = 1; t < L; t++ ) { for( int i = 0; i < K; i++ ) { for( int j = 0; j < K; j++ ) { if( j != i-1 && j != i+1 ) use[j] += luse[i]; } } refreash(); } for
( int i = 1; i < K; i++ ) sum += luse[i]%a; cout<<sum%a<<endl; return 0; } void refreash() { for( int i = 0; i < K; i++ ) { luse[i]=use[i]%a; use[i]=0; } return ; }

K好數(DP)