fix the function count_operations below to get the correct output getting the following results

Input (stdin)

2 2 3

your Output (stdout)

13 33 

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): # Count the number of operations incurred by calling tabulate on each value in the given list results = [] for n in ns: count = 0 for i in range(1, n+1): for j in range(1, n+1): count += 1 count += 1 count += (i*j)%n count += (i//j) results.append(count) return results