寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #D - 禁止された数字

AtCoder Beginner Contestの過去問から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#D - 禁止された数字

atcoder.jp

僕が作成、提出したコードは、以下のとおりです。

/*
AtCoder Problems in C++
#D - 禁止された数字
https://atcoder.jp/contests/abc007/tasks/abc007_4
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
ll f(string s){
  ll ss=s.size();
  vector<vector<vector<ll>>> dp(ss+1,vector<vector<ll>>(2,vector<ll>(2,0)));
  dp[0][0][0]=1;
  for(ll i=0;i<ss;i++){
    ll num=s[i]-48;
    for(ll j=0;j<2;j++){
      for(ll k=0;k<2;k++){
        for(ll l=0;l<=(j ? 9 : num);l++){
          dp[i+1][j || (l<num)][k || l==4 || l==9]+=dp[i][j][k];
        }
      }
    }
  }
  return dp[ss][0][1]+dp[ss][1][1];
}
int main(void){
  ll a,b;
  cin>>a>>b;
  cout<<f(to_string(b))-f(to_string(a-1))<<endl;
  return 0;
} 

AtCoder Beginner Contestは、オンラインジャッジによるプログラミングコンテストです。
日本語と英語に対応していて、週末ごとに実施されているみたいです。
https://practice.contest.atcoder.jp/tutorial
アカウントを登録すれば、誰でも参加できます。