题目描述
学长有n个棒棒糖,他每次最多可以吃k个,请问他有多少种方法来吃棒棒糖
输入
输入两个整数n, k(1 <= n <= 100000) (1 <= k <= 100)
输出
一个正整数,为不同方法数,由于答案可能很大,你需要输出对答案mod100003的结果。
样例输入
5 2
样例输出
8
提示
8种例子如下:
1-1-1-1-1
2-1-1-1
1-2-1-1
1-1-2-1
1-1-1-2
2-2-1
2-1-2
1-2-2
还是一道动态规划的题。
dp[n]代表着n个棒棒糖时有几种方式吃掉。
dp[n]的值是dp[n-1]到dp[n-k]的和。
假设当n=5,k=3时,dp[5]的值就是有4个棒棒糖时吃掉的方式+有3个棒棒糖时吃掉的方式+有2个棒棒糖时吃掉的方式。即dp[5]= dp[4]+dp[3]+dp[2]。
#include <bits/stdc++.h>
using namespace std;
const int mod = 100003;
int dp[100005];
int main(){
memset(dp, 0, sizeof(dp));
int n, k;
cin >> n >> k;
int ans = 0;
dp[0] = 1;
for(int i = 1; i <= n; i++){
int num = 0;
for(int j = 1; j <= k; j++){
if(i-j < 0) break;
num = (num + dp[i-j]) % mod;
}
dp[i] = num;
}
cout << dp[n];
}