29 lines
617 B
C++
29 lines
617 B
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
const int N = 5010;
|
||
int f[N]; // DP结果数组 f[i]含义：第i层台阶共有多少种到达的方法数
|
||
int n, k;
|
||
/*
|
||
4 2
|
||
5
|
||
*/
|
||
int main() {
|
||
cin >> n >> k;
|
||
// 特判K=1
|
||
if (k == 1) {
|
||
cout << 1 << endl;
|
||
exit(0);
|
||
}
|
||
// 初始化 K>=2
|
||
f[1] = 1;
|
||
f[2] = 2;
|
||
|
||
// 状态转移方程
|
||
for (int i = 3; i <= n; i++)
|
||
for (int j = max(i - k, 1); j <= max(1, i - 1); j++)
|
||
f[i] += f[j];
|
||
|
||
// for (int i = 1; i <= n; i++) cout << f[i] << " ";
|
||
cout << f[n] << endl;
|
||
return 0;
|
||
} |