動機

我忘了lowerbound

Problem

Given a non-negative integer x, compute and return the square root of x.

Since the return type is an integer, the decimal digits are truncated, and only the integer part of the result is returned.

Note: You are not allowed to use any built-in exponent function or operator, such as pow(x, 0.5) or x ** 0.5.

 

Example 1:

Input: x = 4Output: 2

Example 2:

Input: x = 8Output: 2Explanation: The square root of 8 is 2.82842..., and since the decimal part is truncated, 2 is returned.

 

Constraints:

  • 0 <= x <= 231 - 1

Sol

所以我直接做bsearch,但要處理loop,loop發生在left等於mid時

class Solution:
    def mySqrt(self, x: int) -> int:
        a,b = 0,x+1 # 猜數字
        while a<b:
            mid = (a+b)//2
            tmp = mid**2
            if tmp == x or mid == a:
                return mid
            elif tmp > x:
                b = mid
            else:
                a = mid
        return a