Sum of ones

#include <bits/stdc++.h>

#define int long long
void init_code() {
    std::ios_base::sync_with_stdio(0);
    std::cin.tie(0);
    #ifndef ONLINE_JUDGE
        freopen("input.txt", "r", stdin);
        freopen("output.txt", "w", stdout);
    #endif
}

using namespace std;

void solve() {
    int n;
    cin >> n;

    n += 1;
    
    int res = 0;
    int step = 2;
    
    bool flag = true;
    
    while (flag) {
        int tmp = 0;

        tmp += (n / step) * (step / 2);    
        //tmp += (n % step) - (step / 2);
        //cout << n % step << '\n';
        int x = (n % step);
        int y = (step / 2);

        //cout << tmp << '\n';
        //cout << x << ' ' << y << '\n';
        //cout << x - y << '\n';
        //cout << '\n';

        tmp += x % y;

        res += tmp;

        if (tmp == 0 || step >= n) {
            flag = false;
        }

        step = step * 2;
    }

    cout << res << '\n';
}


signed main() {
    init_code();
    int t = 1;
    //cin >> t;
    while (t--) {
        solve();
    }
    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: WRONG ANSWER

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER

Test 13

Verdict: WRONG ANSWER

Test 14

Verdict: WRONG ANSWER

Test 15

Verdict: WRONG ANSWER

Test 16

Verdict: WRONG ANSWER

Test 17

Verdict: WRONG ANSWER

Test 18

Verdict: WRONG ANSWER

Test 19

Verdict: WRONG ANSWER

Test 20

Verdict: WRONG ANSWER

Test 21

Verdict: WRONG ANSWER