Problem p02609 (C++)

Buggy submission (status: Runtime Error):
#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (long long i = 0; i < (long long)(n); i++)
#define REP(i, k, n) for (long long i = k; i < (long long)(n); i++)
#define all(a) a.begin(), a.end()
#define pb emplace_back
#define eb emplace_back
#define lb(v, k) (lower_bound(all(v), k) - v.begin())
#define ub(v, k) (upper_bound(all(v), k) - v.begin())
#define fi first
#define se second
#define pi M_PI
#define PQ(T) priority_queue<T>
#define SPQ(T) priority_queue<T, vector<T>, greater<T>>
#define dame(a)                                                                \
  {                                                                            \
    out(a);                                                                    \
    return 0;                                                                  \
  }
#define decimal cout << fixed << setprecision(15);
#define dupli(a) a.erase(unique(all(a)), a.end())
typedef long long ll;
typedef pair<ll, ll> P;
typedef tuple<ll, ll, ll> PP;
typedef tuple<ll, ll, ll, ll> PPP;
typedef multiset<ll> S;
using vi = vector<ll>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using vp = vector<P>;
using vvp = vector<vp>;
using vb = vector<bool>;
using vvb = vector<vb>;
const ll inf = 1001001001001001001;
const ll INF = 1001001001;
const ll mod = 1000000007;
const double eps = 1e-10;
template <class T> bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> void out(T a) { cout << a << '\n'; }
template <class T> void outp(T a) {
  cout << '(' << a.fi << ',' << a.se << ')' << '\n';
}
template <class T> void outvp(T v) {
  rep(i, v.size()) cout << '(' << v[i].fi << ',' << v[i].se << ')';
  cout << '\n';
}
template <class T> void outvvp(T v) { rep(i, v.size()) outvp(v[i]); }
template <class T> void outv(T v) {
  rep(i, v.size()) {
    if (i)
      cout << ' ';
    cout << v[i];
  }
  cout << '\n';
}
template <class T> void outvv(T v) { rep(i, v.size()) outv(v[i]); }
template <class T> bool isin(T x, T l, T r) { return (l) <= (x) && (x) <= (r); }
template <class T> void yesno(T b) {
  if (b)
    out("yes");
  else
    out("no");
}
template <class T> void YesNo(T b) {
  if (b)
    out("Yes");
  else
    out("No");
}
template <class T> void YESNO(T b) {
  if (b)
    out("YES");
  else
    out("NO");
}
template <class T> void noyes(T b) {
  if (b)
    out("no");
  else
    out("yes");
}
template <class T> void NoYes(T b) {
  if (b)
    out("No");
  else
    out("Yes");
}
template <class T> void NOYES(T b) {
  if (b)
    out("NO");
  else
    out("YES");
}
void outs(ll a, ll b) {
  if (a >= inf - 100)
    out(b);
  else
    out(a);
}
ll gcd(ll a, ll b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}
ll modpow(ll a, ll b) {
  ll res = 1;
  a %= mod;
  while (b) {
    if (b & 1)
      res = res * a % mod;
    a = a * a % mod;
    b >>= 1;
  }
  return res;
}
ll g(ll a) { return __builtin_popcount(a); }
vi dp;
ll dfs(int i) {
  if (dp[i] != -1)
    return dp[i];
  if (i == 0) {
    dp[i] = 0;
    return 0;
  }
  dp[i] = dfs(i % g(i)) + 1;
  return dp[i];
}
int main() {
  ll n;
  cin >> n;
  string s;
  cin >> s;
  dp = vi(n + 1, -1);
  for (int i = n; i >= 0; i--)
    dfs(i);
  ll cnt = 0;
  rep(i, n) if (s[i] == '1') cnt++;
  vi ans(n);
  if (cnt) {
    ll d = 0;
    rep(i, n) {
      d *= 2;
      d += s[i] - '0';
      d %= (cnt - 1);
    }
    vi v(n);
    v[n - 1] = 1;
    for (int i = n - 2; i >= 0; i--)
      v[i] = v[i + 1] * 2 % (cnt - 1);
    rep(i, n) if (s[i] == '1') {
      if (cnt == 1)
        ans[i] = 0;
      else
        ans[i] = dp[(d + cnt - 1 - v[i]) % (cnt - 1)] + 1;
    }
  }
  if (cnt != n) {
    ll d = 0;
    rep(i, n) {
      d *= 2;
      d += s[i] - '0';
      d %= (cnt + 1);
    }
    vi v(n);
    v[n - 1] = 1;
    for (int i = n - 2; i >= 0; i--)
      v[i] = v[i + 1] * 2 % (cnt + 1);
    rep(i, n) if (s[i] == '0') { ans[i] = dp[(d + v[i]) % (cnt + 1)] + 1; }
  }
  for (ll x : ans)
    out(x);
}

Fixed submission:
#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (long long i = 0; i < (long long)(n); i++)
#define REP(i, k, n) for (long long i = k; i < (long long)(n); i++)
#define all(a) a.begin(), a.end()
#define pb emplace_back
#define eb emplace_back
#define lb(v, k) (lower_bound(all(v), k) - v.begin())
#define ub(v, k) (upper_bound(all(v), k) - v.begin())
#define fi first
#define se second
#define pi M_PI
#define PQ(T) priority_queue<T>
#define SPQ(T) priority_queue<T, vector<T>, greater<T>>
#define dame(a)                                                                \
  {                                                                            \
    out(a);                                                                    \
    return 0;                                                                  \
  }
#define decimal cout << fixed << setprecision(15);
#define dupli(a) a.erase(unique(all(a)), a.end())
typedef long long ll;
typedef pair<ll, ll> P;
typedef tuple<ll, ll, ll> PP;
typedef tuple<ll, ll, ll, ll> PPP;
typedef multiset<ll> S;
using vi = vector<ll>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using vp = vector<P>;
using vvp = vector<vp>;
using vb = vector<bool>;
using vvb = vector<vb>;
const ll inf = 1001001001001001001;
const ll INF = 1001001001;
const ll mod = 1000000007;
const double eps = 1e-10;
template <class T> bool chmin(T &a, T b) {
  if (a > b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
}
template <class T> void out(T a) { cout << a << '\n'; }
template <class T> void outp(T a) {
  cout << '(' << a.fi << ',' << a.se << ')' << '\n';
}
template <class T> void outvp(T v) {
  rep(i, v.size()) cout << '(' << v[i].fi << ',' << v[i].se << ')';
  cout << '\n';
}
template <class T> void outvvp(T v) { rep(i, v.size()) outvp(v[i]); }
template <class T> void outv(T v) {
  rep(i, v.size()) {
    if (i)
      cout << ' ';
    cout << v[i];
  }
  cout << '\n';
}
template <class T> void outvv(T v) { rep(i, v.size()) outv(v[i]); }
template <class T> bool isin(T x, T l, T r) { return (l) <= (x) && (x) <= (r); }
template <class T> void yesno(T b) {
  if (b)
    out("yes");
  else
    out("no");
}
template <class T> void YesNo(T b) {
  if (b)
    out("Yes");
  else
    out("No");
}
template <class T> void YESNO(T b) {
  if (b)
    out("YES");
  else
    out("NO");
}
template <class T> void noyes(T b) {
  if (b)
    out("no");
  else
    out("yes");
}
template <class T> void NoYes(T b) {
  if (b)
    out("No");
  else
    out("Yes");
}
template <class T> void NOYES(T b) {
  if (b)
    out("NO");
  else
    out("YES");
}
void outs(ll a, ll b) {
  if (a >= inf - 100)
    out(b);
  else
    out(a);
}
ll gcd(ll a, ll b) {
  if (b == 0)
    return a;
  return gcd(b, a % b);
}
ll modpow(ll a, ll b) {
  ll res = 1;
  a %= mod;
  while (b) {
    if (b & 1)
      res = res * a % mod;
    a = a * a % mod;
    b >>= 1;
  }
  return res;
}
ll g(ll a) { return __builtin_popcount(a); }
vi dp;
ll dfs(int i) {
  if (dp[i] != -1)
    return dp[i];
  if (i == 0) {
    dp[i] = 0;
    return 0;
  }
  dp[i] = dfs(i % g(i)) + 1;
  return dp[i];
}
int main() {
  ll n;
  cin >> n;
  string s;
  cin >> s;
  dp = vi(n + 1, -1);
  for (int i = n; i >= 0; i--)
    dfs(i);
  ll cnt = 0;
  rep(i, n) if (s[i] == '1') cnt++;
  vi ans(n);
  if (cnt > 1) {
    ll d = 0;
    rep(i, n) {
      d *= 2;
      d += s[i] - '0';
      d %= (cnt - 1);
    }
    vi v(n);
    v[n - 1] = 1;
    for (int i = n - 2; i >= 0; i--)
      v[i] = v[i + 1] * 2 % (cnt - 1);
    rep(i, n) if (s[i] == '1') {
      if (cnt == 1)
        ans[i] = 0;
      else
        ans[i] = dp[(d + cnt - 1 - v[i]) % (cnt - 1)] + 1;
    }
  }
  if (cnt != n) {
    ll d = 0;
    rep(i, n) {
      d *= 2;
      d += s[i] - '0';
      d %= (cnt + 1);
    }
    vi v(n);
    v[n - 1] = 1;
    for (int i = n - 2; i >= 0; i--)
      v[i] = v[i + 1] * 2 % (cnt + 1);
    rep(i, n) if (s[i] == '0') { ans[i] = dp[(d + v[i]) % (cnt + 1)] + 1; }
  }
  for (ll x : ans)
    out(x);
}