Problem p03178 (C++)

Buggy submission (status: Runtime Error):
/*
これを入れて実行
g++ code.cpp
./a.out
 */

#include <algorithm>
#include <bitset>
#include <cmath>
#include <deque>
#include <iomanip>
#include <iostream>
#include <map>
#include <math.h>
#include <queue>
#include <set>
#include <stdio.h>
#include <string>
#include <tuple>
#include <unordered_map>
#include <utility>
#include <vector>

using namespace std;
typedef long long ll;
typedef long double ld;

int dy4[4] = {-1, 0, +1, 0};
int dx4[4] = {0, +1, 0, -1};
int dy8[8] = {-1, -1, 0, 1, 1, 1, 0, -1};
int dx8[8] = {0, 1, 1, 1, 0, -1, -1, -1};

const long long INF = 1LL << 60;
const ll MOD = 1e9 + 7;

bool greaterSecond(const pair<int, int> &f, const pair<int, int> &s) {
  return f.second > s.second;
}

ll gcd(ll a, ll b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}

ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }

ll nCr(ll n, ll r) {
  if (r == 0 || r == n) {
    return 1;
  } else if (r == 1) {
    return n;
  }
  return (nCr(n - 1, r) + nCr(n - 1, r - 1));
}

ll nPr(ll n, ll r) {
  r = n - r;
  ll ret = 1;
  for (ll i = n; i >= r + 1; i--)
    ret *= i;
  return ret;
}

//-----------------------ここから-----------
string k;
int D;
ll memo[10010][2][110];

ll dp(int i, int f, int d) {
  if (i == k.size())
    return d == 0 ? 1 : 0;
  if (memo[i][f][d] != -1)
    return memo[i][f][d];

  ll res = 0;
  int now = k[i] - '0';

  for (int di = 0; di <= (f ? 9 : now); di++) {
    res += dp(i + 1, (f || (di < now)), (d + di) % D) % MOD;
    res %= MOD;
  }

  return memo[i][f][d] = res;
}

void init() {
  for (int i = 0; i <= k.size(); i++) {
    for (int j = 0; j < 2; j++) {
      for (int l = 0; l <= D; l++) {
        memo[i][j][l] = -1;
      }
    }
  }
}

int main(void) {
  cin >> D >> k;

  init();
  ll ans = dp(0, 0, 0) + ((-1) % MOD);
  if (ans < 0)
    ans += MOD;
  cout << ans % MOD << endl;
}

Fixed submission:
/*
これを入れて実行
g++ code.cpp
./a.out
 */

#include <algorithm>
#include <bitset>
#include <cmath>
#include <deque>
#include <iomanip>
#include <iostream>
#include <map>
#include <math.h>
#include <queue>
#include <set>
#include <stdio.h>
#include <string>
#include <tuple>
#include <unordered_map>
#include <utility>
#include <vector>

using namespace std;
typedef long long ll;
typedef long double ld;

int dy4[4] = {-1, 0, +1, 0};
int dx4[4] = {0, +1, 0, -1};
int dy8[8] = {-1, -1, 0, 1, 1, 1, 0, -1};
int dx8[8] = {0, 1, 1, 1, 0, -1, -1, -1};

const long long INF = 1LL << 60;
const ll MOD = 1e9 + 7;

bool greaterSecond(const pair<int, int> &f, const pair<int, int> &s) {
  return f.second > s.second;
}

ll gcd(ll a, ll b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}

ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }

ll nCr(ll n, ll r) {
  if (r == 0 || r == n) {
    return 1;
  } else if (r == 1) {
    return n;
  }
  return (nCr(n - 1, r) + nCr(n - 1, r - 1));
}

ll nPr(ll n, ll r) {
  r = n - r;
  ll ret = 1;
  for (ll i = n; i >= r + 1; i--)
    ret *= i;
  return ret;
}

//-----------------------ここから-----------
string k;
int D;
ll memo[10010][2][110];

ll dp(int i, int f, int d) {
  if (i == k.size())
    return d == 0 ? 1 : 0;
  if (memo[i][f][d] != -1)
    return memo[i][f][d];

  ll res = 0;
  int now = k[i] - '0';

  for (int di = 0; di <= (f ? 9 : now); di++) {
    res += dp(i + 1, (f || (di < now)), (d + di) % D) % MOD;
    res %= MOD;
  }

  return memo[i][f][d] = res;
}

void init() {
  for (int i = 0; i <= k.size(); i++) {
    for (int j = 0; j < 2; j++) {
      for (int l = 0; l <= D; l++) {
        memo[i][j][l] = -1;
      }
    }
  }
}

int main(void) {
  cin >> k >> D;

  init();
  ll ans = dp(0, 0, 0) + ((-1) % MOD);
  if (ans < 0)
    ans += MOD;
  cout << ans % MOD << endl;
}