why keep getting for the wrong answer for the code below

Input (stdin)

2 2 3

Your Output (stdout)

4 11 

Expected Output

38 183

def tab(n): tbl = [] i = 0 while i < n: tbl.append(foo(n, i)) i = i + 1 return tbl def foo(n, r): ub = 1 count = 0 for i in range(1, n+1): for j in range(ub): count = count + bar(j, r) ub = 2 * ub return count def bar(val, r): count = 0 ncount = 0 v = val for j in range(2 * r): if v % 2 == 1: count = count + 1 else: ncount = ncount + 1 v = v // 2 if count == r: return 1 else: return 0

def count_operations(ns): results = [] for n in ns: count = 0 for i in range(n): count += foo(n, i) results.append(count) return results

if __name__ == '__main__': fptr = open(os.environ['OUTPUT_PATH'], 'w')

k = int(input().strip())

ns = list(map(int, input().rstrip().split()))

result = count_operations(ns)

fptr.write(' '.join(map(str, result))) fptr.write(' ')

fptr.close()