EOJ3531. 定西（爬楼梯问题plus版）

单点时限: 1.0 sec

内存限制: 256 MB

这么多年你一个人一直在走
方向和天气的节奏会让你忧愁
你说你遇见了一大堆奇怪的人
他们看上去好像都比你开心
——李志《定西》

这首歌的吉他节奏总感觉是在致敬《加州旅馆》，前奏又像葫芦娃里面在蛇精洞是的配乐

一个人走走了很多年，发现自己走到了一个很长的，年久失修的楼梯面前。年久失修的意思就是，有 k 个台阶坏了，没法走。

楼梯一共有 n 层，你一次能上一阶、两阶或三阶台阶，请问，你从楼梯底部 (0 开始) 走到楼梯顶部，共有多少种走法。

输入格式

输入数据共两行，第一行包含两个自然数 n (1≤n≤100) 和 k (0≤k<n)，第二行包含 k 个自然数 Xi (1≤Xi≤n)，数字之间用一个空格隔开，表示损坏的台阶的序号（从楼梯底部到楼梯顶部，台阶序号依次为 1 到 n）。

输出格式

输出数据仅包含一个整数，表示所有可行走法的总数。

样例

input

5 2
2 4

output

2

#include <iostream>
#include <bits/stdc++.h>
#include <math.h>
using namespace std;

int main() {
	int temp;
	int dp[100] = {0};	
	int TargetStep;	 // n
	int BrokenNum;	 // k
	set<int> BrokenSteps;

	// 输入
	cin >> TargetStep >> BrokenNum;
	for (int j = 0; j < BrokenNum; j++) {
		cin >> temp;
		BrokenSteps.insert(temp);
	}

	// 其中dp[n]代表：第n阶台阶上法总数。若0123阶中出现了broken，则这个台阶的上法数=0.
	dp[0] = 1;
	dp[1] = BrokenSteps.find(1) != BrokenSteps.end() ? 0 : 1;
	dp[2] = BrokenSteps.find(2) != BrokenSteps.end() ? 0 : dp[0] + dp[1] ;
	dp[3] = dp[2] + dp[1] + dp[0];

	// 输出
	for (int j = 0; j <= TargetStep; j++) {
		if (BrokenSteps.find(j) != BrokenSteps.end()) {	// 当前阶梯破损
			dp[j] = 0;
		} else {
			if (j >= 4)
				dp[j] = dp[j - 1] + dp[j - 2] + dp[j - 3];
		}
	}
	cout << dp[TargetStep];
    return 0;
}