zukaABCをコツコツ解いていきます。
本記事は，管理人の競技プロ精進日記としてログを取ったものです。モチベーションを爆上げするために，積極的にアウトプットしていく作戦です。これから競技プログラミングを始めようと考えている人や，なんとなく敷居が高いと感じている人の参考になれば嬉しく思います。その他の記事は以下をご覧ください。
目次
本記事の概要
Atcoderで初心者用のコンテストとして開催されているAtcoder Beginner Contest（通称ABC）を解いていくものです。今回はABC129-C「Typical Stairs」です。
実装
#include <bits/stdc++.h>
#define _GLIBCXX_DEBUG
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define repi(i, a, b) for (int i = (int)(a); i < (int)(b); i++)
using namespace std;
int N=0, M=0;
// 割る数の定義
long long MOD = pow(10, 9)+7;
int main(){
cin >> N >> M;
// 階段が壊れているかどうかのflag
vector<bool> flag(N+1, true);
// DPテーブル。漸化式の計算結果を保存しておく用
vector<long long> DP(N+1, 0);
rep(i, M){
int a = 0;
cin >> a;
flag[a] = false;
}
// 今いる場所は1通りと考えることで辻褄が合う
DP[0] = 1;
// 1段目が壊れていれば0のまま，壊れていなければ1で初期化する
if (flag[1]==true) DP[1]=1;
// MODで割ったあまりを格納していけば良い
rep(i, N-1){
if (flag[i+2]==true) DP[i+2] = (DP[i+1]+DP[i]) % MOD;
}
// DPテーブルは0段目が0インデックスなのでN段目はNインデックス
cout << DP[N] << endl;
}ポイント
今回おさえるべき内容
DPを使ったflag管理を身につける
DPを使った基本的な問題です。最初にflagが全部trueという訳では無いという点で，少しだけ応用問題になっています。とはいえ，簡単な漸化式の問題です。
コメント