Sum of ones

import math

nums = []
nums.append(0)
nums.append(1)
i = 2
while 2 ** i < 10 ** 15 + 5:
    nums.append(nums[-1] * 2 + 2 ** (i - 1))
    i += 1


n = int(input())
res = 0
while n > 0:
    k = int(math.log(n + 1, 2))
    res += nums[k] + (n + 1 - 2 ** k)
    n -= 2 ** k

print(res)

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED