Sqrt Leetcode Solution Python