Problem p02609 (C++)

Buggy submission (status: Runtime Error):
/// #pragma GCC optimize("O3,unroll-loops")
/// #pragma GCC target("avx,avx2,fma")

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
#define ll long long
#define ull unsigned long long
#define ld long double
#define inf 2000000000
#define infLL 2000000000000000000
#define MAX5 100005
#define MAX6 1000006
#define MAX7 10000007
#define sf(a) scanf("%d", &a)
#define sfl(a) scanf("%lld", &a)
#define pf(a) printf("%d\n", a)
#define pfl(a) printf("%lld\n", a)
#define Case(t) printf("Case %d: ", t)
#define pii pair<int, int>
#define mod 1000000007
#define Mod 998244353
#define PI acos(-1.0)
#define eps 1e-9
#define mem(a, b) memset(a, b, sizeof(a))
#define FASTIO ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);

#define error(args...)                                                         \
  {                                                                            \
    string _s = #args;                                                         \
    replace(_s.begin(), _s.end(), ',', ' ');                                   \
    stringstream _ss(_s);                                                      \
    istream_iterator<string> _it(_ss);                                         \
    err(_it, args);                                                            \
    cerr << '\n';                                                              \
  }

inline int Set(ll N, ll pos) { return N = N | (1LL << pos); }
inline int Reset(ll N, ll pos) { return N = N & ~(1LL << pos); }
inline bool Check(ll N, ll pos) { return (bool)(N & (1LL << pos)); }

inline bool Equal(ld x, ld y) { return fabs(x - y) < eps; } /// x==y
inline bool Greater(ld x, ld y) { return (x - eps) > y; }   /// x>y
inline bool Lesser(ld x, ld y) { return (x + eps) < y; }    /// x<y

inline void normal(ll &a) {
  a %= mod;
  (a < 0) && (a += mod);
}
inline ll modMul(ll a, ll b) {
  a %= mod, b %= mod;
  normal(a), normal(b);
  return (a * b) % mod;
}
inline ll modAdd(ll a, ll b) {
  a %= mod, b %= mod;
  normal(a), normal(b);
  return (a + b) % mod;
}
inline ll modSub(ll a, ll b) {
  a %= mod, b %= mod;
  normal(a), normal(b);
  a -= b;
  normal(a);
  return a;
}
inline ll modPow(ll b, ll p) {
  ll r = 1;
  while (p) {
    if (p & 1LL)
      r = modMul(r, b);
    b = modMul(b, b);
    p >>= 1LL;
  }
  return r;
}
inline ll modInverse(ll a) { return modPow(a, mod - 2); }
inline ll modDiv(ll a, ll b) { return modMul(a, modInverse(b)); }

/// Memory Efficient
inline ll fast_exp(ll x, ll n, ll m) {
  ll res = 1;
  while (n > 0) {
    if (n % 2 == 1)
      res = (res * x) % m;
    x = (x * x) % m;
    n /= 2;
  }
  return res % m;
}

// int dx[] = {+0, +0, -1, +1}; ///Up-down, Left-Right
// int dy[] = {+1, -1, +0, +0};
// int dx[] = {+0,+0,+1,-1,-1,+1,-1,+1}; ///King's Move
// int dy[] = {-1,+1,+0,+0,+1,+1,-1,-1};
// int dx[] = {-2, -2, -1, -1,  1,  1,  2,  2}; ///Knight's Move
// int dy[] = {-1,  1, -2,  2, -2,  2, -1,  1};

void err(istream_iterator<string> it) {}
template <typename T, typename... Args>
void err(istream_iterator<string> it, T a, Args... args) {
  cerr << *it << " = " << a << " ";
  err(++it, args...);
}

inline void Time() {
  cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
}

typedef tree<int, null_type, less<int>, rb_tree_tag,
             tree_order_statistics_node_update>
    orderedSet;
typedef tree<int, null_type, greater<int>, rb_tree_tag,
             tree_order_statistics_node_update>
    orderedSet1;

const int N = 200001;

ll solve(int x) {
  int ans = 0;
  while (x > 0) {
    int cnt = __builtin_popcount(x);
    x = x % cnt;
    ans++;
  }
  return ans;
}

int main() {
  // freopen("in.txt","r",stdin);
  // freopen("out.txt","w",stdout);

  FASTIO;
  int n;
  cin >> n;
  string s;
  cin >> s;
  int one = 0;
  for (int i = 0; i < n; i++)
    if (s[i] == '1')
      one++;
  int sum0 = 0, sum1 = 0;
  for (int i = 0; i < n; i++) {
    if (s[i] == '0')
      continue;
    if (one > 0)
      sum0 = (sum0 + fast_exp(2, n - 1 - i, one - 1)) % (one - 1);
    sum1 = (sum1 + fast_exp(2, n - 1 - i, one + 1)) % (one + 1);
  }
  /// error(sum0, sum1);
  ll ans;
  for (int i = 0; i < n; i++) {
    ll sum = 0;
    if (s[i] == '1') {
      sum = sum0;
      if (one > 1) {
        sum = (sum - fast_exp(2, n - 1 - i, one - 1)) % (one - 1);
        if (sum < 0)
          sum += (one - 1);
        ans = 1 + solve(sum);
      } else
        ans = 0;
    } else {
      sum = sum1;
      sum = (sum + fast_exp(2, n - 1 - i, one + 1)) % (one + 1);
      ans = 1 + solve(sum);
    }
    cout << ans << endl;
  }

  /// Time();
  return 0;
}


Fixed submission:
/// #pragma GCC optimize("O3,unroll-loops")
/// #pragma GCC target("avx,avx2,fma")

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
#define ll long long
#define ull unsigned long long
#define ld long double
#define inf 2000000000
#define infLL 2000000000000000000
#define MAX5 100005
#define MAX6 1000006
#define MAX7 10000007
#define sf(a) scanf("%d", &a)
#define sfl(a) scanf("%lld", &a)
#define pf(a) printf("%d\n", a)
#define pfl(a) printf("%lld\n", a)
#define Case(t) printf("Case %d: ", t)
#define pii pair<int, int>
#define mod 1000000007
#define Mod 998244353
#define PI acos(-1.0)
#define eps 1e-9
#define mem(a, b) memset(a, b, sizeof(a))
#define FASTIO ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);

#define error(args...)                                                         \
  {                                                                            \
    string _s = #args;                                                         \
    replace(_s.begin(), _s.end(), ',', ' ');                                   \
    stringstream _ss(_s);                                                      \
    istream_iterator<string> _it(_ss);                                         \
    err(_it, args);                                                            \
    cerr << '\n';                                                              \
  }

inline int Set(ll N, ll pos) { return N = N | (1LL << pos); }
inline int Reset(ll N, ll pos) { return N = N & ~(1LL << pos); }
inline bool Check(ll N, ll pos) { return (bool)(N & (1LL << pos)); }

inline bool Equal(ld x, ld y) { return fabs(x - y) < eps; } /// x==y
inline bool Greater(ld x, ld y) { return (x - eps) > y; }   /// x>y
inline bool Lesser(ld x, ld y) { return (x + eps) < y; }    /// x<y

inline void normal(ll &a) {
  a %= mod;
  (a < 0) && (a += mod);
}
inline ll modMul(ll a, ll b) {
  a %= mod, b %= mod;
  normal(a), normal(b);
  return (a * b) % mod;
}
inline ll modAdd(ll a, ll b) {
  a %= mod, b %= mod;
  normal(a), normal(b);
  return (a + b) % mod;
}
inline ll modSub(ll a, ll b) {
  a %= mod, b %= mod;
  normal(a), normal(b);
  a -= b;
  normal(a);
  return a;
}
inline ll modPow(ll b, ll p) {
  ll r = 1;
  while (p) {
    if (p & 1LL)
      r = modMul(r, b);
    b = modMul(b, b);
    p >>= 1LL;
  }
  return r;
}
inline ll modInverse(ll a) { return modPow(a, mod - 2); }
inline ll modDiv(ll a, ll b) { return modMul(a, modInverse(b)); }

/// Memory Efficient
inline ll fast_exp(ll x, ll n, ll m) {
  ll res = 1;
  while (n > 0) {
    if (n % 2 == 1)
      res = (res * x) % m;
    x = (x * x) % m;
    n /= 2;
  }
  return res % m;
}

// int dx[] = {+0, +0, -1, +1}; ///Up-down, Left-Right
// int dy[] = {+1, -1, +0, +0};
// int dx[] = {+0,+0,+1,-1,-1,+1,-1,+1}; ///King's Move
// int dy[] = {-1,+1,+0,+0,+1,+1,-1,-1};
// int dx[] = {-2, -2, -1, -1,  1,  1,  2,  2}; ///Knight's Move
// int dy[] = {-1,  1, -2,  2, -2,  2, -1,  1};

void err(istream_iterator<string> it) {}
template <typename T, typename... Args>
void err(istream_iterator<string> it, T a, Args... args) {
  cerr << *it << " = " << a << " ";
  err(++it, args...);
}

inline void Time() {
  cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
}

typedef tree<int, null_type, less<int>, rb_tree_tag,
             tree_order_statistics_node_update>
    orderedSet;
typedef tree<int, null_type, greater<int>, rb_tree_tag,
             tree_order_statistics_node_update>
    orderedSet1;

const int N = 200001;

ll solve(int x) {
  int ans = 0;
  while (x > 0) {
    int cnt = __builtin_popcount(x);
    x = x % cnt;
    ans++;
  }
  return ans;
}

int main() {
  // freopen("in.txt","r",stdin);
  // freopen("out.txt","w",stdout);

  FASTIO;
  int n;
  cin >> n;
  string s;
  cin >> s;
  int one = 0;
  for (int i = 0; i < n; i++)
    if (s[i] == '1')
      one++;
  int sum0 = 0, sum1 = 0;
  for (int i = 0; i < n; i++) {
    if (s[i] == '0')
      continue;
    if (one > 1)
      sum0 = (sum0 + fast_exp(2, n - 1 - i, one - 1)) % (one - 1);
    sum1 = (sum1 + fast_exp(2, n - 1 - i, one + 1)) % (one + 1);
  }
  /// error(sum0, sum1);
  ll ans;
  for (int i = 0; i < n; i++) {
    ll sum = 0;
    if (s[i] == '1') {
      sum = sum0;
      if (one > 1) {
        sum = (sum - fast_exp(2, n - 1 - i, one - 1)) % (one - 1);
        if (sum < 0)
          sum += (one - 1);
        ans = 1 + solve(sum);
      } else
        ans = 0;
    } else {
      sum = sum1;
      sum = (sum + fast_exp(2, n - 1 - i, one + 1)) % (one + 1);
      ans = 1 + solve(sum);
    }
    cout << ans << endl;
  }

  /// Time();
  return 0;
}