K好數(DP)
阿新 • • 發佈:2017-08-21
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)