【C++题解】走楼梯-高阶版

$Part$ $1$ 读题

题目描述

$tywz$的同学都是具有拼搏精神和团队精神的，通过不断的努力，下头男$why$进入了$tywz$信竞团队，在知行楼$5$楼有一个神秘宝座，要到达宝座需要爬数级台阶，只见有一个$N$级台阶，编号$0$-$N$，他们现在在台阶下面（姑且认为为第$0$级），而宝座就在顶端（第$N$级），正当大家准备冲上楼梯时——

下头男$why$：“慢着，楼梯上有炸弹！！！”

你：“什么？$w\ast f$！！！”

只见台阶上有$m$个$sa$炸弹，安在$m$个不同台阶上，如果你踩到安有炸弹的台阶，你就真成$sa$了。爬楼梯的时候一次只能爬$1$、$2$级或$3$级台阶。

“好吧，让我们算算走上楼梯且没踩到$sa$炸弹的方案总数吧！”

输入格式

第一行 两个整数 $N$，$M$。$M<N<1000$

第二行 $m$个整数，第$i$个整数为第$i$个炸弹所在台阶编号$Bi$

输出格式

走上台阶且没被炸成$sa$的方案总数，由于结果可能过大，输出结果对$12580$取模。

如果变成$sa$（非神犇）为必然事件输出$-1$。

输入样例

5 1
4

输出样例

6

样例说明

样例路线：
$0->1->2->3->5$

$0->2->3->5$

$0->1->3->5$

$0->3->5$

$0->2->5$

$0->1->2->5$

数据范围与提示

对于$30\%$的数据 ：$M=0$；

对于$40\%$的数据 ：$N\le 20$，$M\le 10$；

对于$80\%$的数据 ：$N\le 100$，$M\le 50$；

对于$100\%$的数据 ：$N\le 1000$，$M\le 300$，$0\le Bi\le N$；

$Part$ $2$ 思路

定义数组$f$，$f[i]$表示到达第$i$个台阶的方案数，可推出公式$f[i]=f[i-1]+f[i-2]+f[i-3]$，最终答案为$f[n]$。

但是要注意！！！

有一些台阶不可以踩，我们需要标记这些台阶（代码中的数组$b$），同时要把到达这些台阶的方案数清零。

$Part$ $3$ 代码

#include<bits/stdc++.h>
using namespace std;
int a[1005],n,m,x;
bool b[1005];
int main(){
	cin>>m>>n;
	for(int i=0;i<n;i++){
		cin>>x;
		b[x]=1;
	}
	if(b[0]==1)a[0]=0;
	else a[0]=1;
	for(int i=1;i<=m;i++)if(!b[i])for(int j=max(0,i-3);j<i;j++)a[i]=(a[i]+a[j])%12580;
	if(a[m]==0)a[m]=-1;
	cout<<a[m];
	return 0;
}

听完后，是不是觉得很简单呢？赶快自己去试一下吧！！！