3531. 定西

3531. 定西

#include<bits/stdc++.h>
using namespace std;
int main() {
    int n,k,temp;
    cin>>n>>k;
    vector<bool> stairs(k+1,true);
    for(int i=1;i<=k;++i){
        cin>>temp;
        stairs[temp]=false;
    }
    vector<int> dp(105,0);
    dp[0]=1;
    if(stairs[1]) dp[1]=1;
    if(stairs[2]) dp[2]=dp[1]+dp[0];
    for(int i=3;i<=n;i++){
        if(stairs[i]){
            dp[i]=dp[i-1]+dp[i-2]+dp[i-3];
        }
    }
    cout<<dp[n];
    return 0;
}