力扣刷题
int mySqrt(int x){
if (x == 0) { // 特判:如果 x 为 0,直接返回 0
return 0;
}
int left = 1, right = x; // 区间左右端点为 1 和 x
while (left <= right) { // 二分查找
int mid = left + (right - left) / 2; // 取区间的中点
if (mid <= x / mid && (mid + 1) > x / (mid + 1)) { // 找到 x 的平方根;mid <= x / mid 用于判断mid * mid 是否小于等于x。(mid + 1) > x / (mid + 1) 用于判断 (mid + 1) * (mid + 1) 是否大于x。如果这两个条件都满足,说明mid正好是x的平方根,此时函数返回mid即可。
return mid;
} else if (mid > x / mid) { // 如果 mid * mid > x,说明平方根在左边区间
right = mid - 1;
} else { // 否则平方根在右边区间
left = mid + 1;
}
}
return 0; // 如果没有找到,返回 0
}
具体思路:
1. 如果输入的x为0,则直接返回0;
2. 使用二分查找的方式来寻找x的平方根;
3. 初始化区间的左右端点left和right为1和x;
4. 在循环内部,取区间的中点mid,分别判断mid * mid与x的大小关系,根据判断结果更新left或right;
5. 循环继续直到left > right,此时说明没有找到x的平方根;
6. 如果找到了x的平方根,直接返回mid即可。