Python算法例4 求平方根

1. 问题描述

实现int sqrt(int x)函数,计算并返回x的平方根。

2. 问题示例

sqrt(3)=1;sqrt(4)=2;sqrt(5)=2;sqrt(17)=4。

3.代码实现

要实现计算整数x的平方根函数sqrt(x),可以使用二分查找法。

首先,我们定义一个变量left = 0用来表示搜索区间的左边界,以及一个变量right = x用来表示搜索区间的右边界。初始时,我们将搜索区间设置为[0, x]。

然后,我们可以进入一个循环,直到搜索区间收敛到一个整数值。在每次循环中,我们首先计算搜索区间的中间值mid = (left + right) // 2,然后将mid的平方与x进行比较。

  • 如果mid的平方等于x,那么mid就是x的平方根,直接返回mid。
  • 如果mid的平方小于x,那么x的平方根应该在[mid+1, right]范围内,将left更新为mid+1。
  • 如果mid的平方大于x,那么x的平方根应该在[left, mid-1]范围内,将right更新为mid-1。

当搜索区间收敛到一个整数值时,返回该整数值作为x的平方根。

python 复制代码
def sqrt(x):
    if x == 0 or x == 1:
        return x

    left, right = 0, x
    while left <= right:
        mid = (left + right) // 2
        if mid * mid == x:
            return mid
        elif mid * mid < x:
            left = mid + 1
        else:
            right = mid - 1

    return right


print(sqrt(3))  # 输出: 1
print(sqrt(4))  # 输出: 2
print(sqrt(5))  # 输出: 2
print(sqrt(17))  # 输出: 4
python 复制代码
class Solution:
    def sqrt(self,x):
        left = 0
        right = x
        while left <= right:
            mid = (left + right) // 2
            if mid * mid == x:
                return mid
            elif mid * mid < x:
                left = mid + 1
            else:
                right = mid - 1
        return right

if __name__ == '__main__':
    temp = Solution()
    x1 = 5
    x2 = 17
    print(("输入:"+str(x1)))
    print(("输出:"+str(temp.sqrt(x1))))
    print(("输入:" + str(x2)))
    print(("输出:" + str(temp.sqrt(x2))))
相关推荐
吃着火锅x唱着歌13 小时前
LeetCode 1446.连续字符
算法·leetcode·职场和发展
愚润求学13 小时前
【贪心算法】day10
c++·算法·leetcode·贪心算法
吴秋霖13 小时前
主流反爬虫、反作弊防护与风控对抗手段
爬虫·算法·反爬虫技术
java1234_小锋14 小时前
Scikit-learn Python机器学习 - 分类算法 - K-近邻(KNN)算法
python·算法·机器学习
智者知已应修善业14 小时前
【矩阵找最大小所在位置】2022-11-13
c语言·c++·经验分享·笔记·算法·矩阵
shan&cen14 小时前
Day04 前缀和&差分 1109. 航班预订统计 、304. 二维区域和检索 - 矩阵不可变
java·数据结构·算法
手握风云-14 小时前
回溯剪枝的 “减法艺术”:化解超时危机的 “救命稻草”(二)
算法·机器学习·剪枝
QiZhang | UESTC14 小时前
JAVA算法练习题day11
java·开发语言·python·算法·hot100
屁股割了还要学14 小时前
【数据结构入门】排序算法(4)归并排序
c语言·数据结构·学习·算法·排序算法
努力学习的小廉14 小时前
我爱学算法之—— 位运算(上)
c++·算法