def complexity(n): total = 0 k = 0 while total < n: total += k ** 2 k += 1 return k
I was thinking O(n*