高频算法面试题(二十)- 求平方根

717 阅读2分钟

「这是我参与11月更文挑战的第 10 天,活动详情查看:2021最后一次更文挑战

刷算法题,从来不是为了记题,而是练习把实际的问题抽象成具体的数据结构或算法模型,然后利用对应的数据结构或算法模型来进行解题。个人觉得,带着这种思维刷题,不仅能解决面试问题,也能更多的学会在日常工作中思考,如何将实际的场景抽象成相应的算法模型,从而提高代码的质量和性能

求平方根

题目来源LeetCode-69. Sqrt(x)

题目描述

给你一个非负整数 x ,计算并返回 x 的 算术平方根

由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去

注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5

示例

示例 1

输入:x = 4
输出:2

示例 2

输入:x = 8
输出:2
解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去

提示:

  • 0 <= x <= 2^31 - 1

解题

思路

像求平方根、中位数、有序数组中查找目标值这些,都适合用二分法来进行求解

这道题是二分的变形题,求一个数字的平方根,其实就等于在从0 ~ x之间找一个数字,使得该数字的平方等于x,显然0~x之间的数字是有序的,完美契合二分查找的条件

题目也就转换成了在0 ~ x之间找目标数字,用标准的二分解题即可。直接看代码

后边的题会进行分类的刷,感觉这样比较容易分块掌握,零碎的刷,适合查漏补缺。所以后边会先针对二叉树的题进行刷,然后是深度优先、广度优先、回溯、贪心、动态规划

代码

//非递归实现
func mySqrt(x int) int {
    if x == 1 || x == 0 {
		return x
	}
	if x <= 0 {
		return -1
	}

	left,  right := 0,  x
	res := -1
	for left <= right {
		mid := left + (right-left)/2
		if mid * mid <= x {
			res = mid
			left = mid + 1
		} else {
			right = mid - 1
		}
	}
	
	return res
}