题目:台阶问题
题目描述
有N级的台阶,你一开始在底部,每次可以向上迈最多K级台阶(最少1级),问到达第N级台阶有多少种不同方式。
输入输出格式
输入格式:
两个正整数N,K。
输出格式:
一个正整数,为不同方式数,由于答案可能很大,你需要输出ansmod100003后的结果。
输入输出样例
输入样例#1:
5 2
输出样例#1:
8
说明
对于20%的数据,有N≤10,K≤3;
对于40%的数据,有N≤1000;
对于100%的数据,有N≤100000,K≤100。
分析:
假设k固定为2,我们对等于2时的情况进行一定的推导,比如当最高阶为1时,ans为1。当最高阶为2时,ans为2。当最高阶为3时,ans为3。当最高阶为4时,ans为5。当最高阶为4时,ans为8。很显然最高阶当大于等于3时,每一项的值等于前2项的值的和。
同理可得,当台阶数大于等于k时,每一项的值等于前k项的值得和,这里k的取值范围大于2。
k等于1时显然得出ans为1。
因此,这样也求得这样解题的关键。
当然还有问题在于,怎样得到前k项的值的和,以及它们各项的值。这里我们不难发现2点,(注:当k大于3时),第1点就是最高阶为1时恒为1,最高阶为2时恒为2。第2点就是3<=台阶数<=k时,每一阶的ans等于前2阶的和加1。
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,k;
int sum;
while(cin>>n>>k){//这里没按照题意多组输入,但也没什么问题
if(k==1) printf("1\n");//等于k==1时为1
else{
int *p=new int [n];
p[0]=1;p[1]=2;sum=3;
for(int i=2;i<k;++i)
p[i]=(sum+1)%100003,sum+=p[i];//求出小于k每一项的值
if(n>k){//当n大于k时,大于k的每一项等于前k项的和,sum为前k项的和
p[k]=sum%100003;
for(int i=k+1;i<n;++i){
sum-=p[i-k-1];
p[i]=(sum+p[i-1]+100003)%100003;//进行相减得到的sum可能为负数,加100003消除负数影响
sum=p[i];
}
printf("%d\n",p[n-1]%100003);
}
else printf("%d\n",p[n-1]%100003);//n小于k,直接输出。
delete [] p;
}
}
return 0;
}