Sum of ones

n = int(input())

def number_of_ones(n):
    if n == 0:
        return 0

    m = 0
    while n > 1:
        n = n >> 1
        m += 1
    temp = 1 << m
    while n < m:
        temp = temp >> 1
        m -= 1

    if n == ((1 << (m+1)) -1):
        return (m+1)*(1<<m)

    n = n-(1<<m)
    return n+1 + number_of_ones(n) + m*(1<<(m-1))

print(number_of_ones(n))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 4

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 5

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 6

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 7

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 8

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 9

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 10

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 11

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 12

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 13

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 14

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 15

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 16

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 17

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 18

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 19

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 20

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...

Test 21

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 22, in <module>
    print(...