master-algorithms-py/searching/sqrt_x.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# author: bt3gl


def sqrt(x) -> int:
    
        if x < 2:
            return x
        
        left, right = 2, x // 2
        
        while left <= right:
            
            pivot = left + (right - left) // 2
            num = pivot * pivot
            
            if num > x:
                right = pivot - 1
                
            elif num < x:
                left = pivot + 1
                
            else:
                return pivot
            
        return right