【LeetCode剑指offer#06】实现pow函数、计算x的平方根

实现pow函数

实现 pow(x, n) ,即计算 x 的整数 n 次幂函数(即,xn )。

示例 1:

输入:x = 2.00000, n = 10
输出:1024.00000

示例 2:

输入:x = 2.10000, n = 3
输出:9.26100

示例 3:

输入:x = 2.00000, n = -2
输出:0.25000
解释:2-2 = 1/22 = 1/4 = 0.25

代码

class Solution {
public:
    double myPow(double x, int n) {
        if (n == 0) return 1.0; // 特殊情况处理

        long long N = n; // 防止将 int 型 n 转为负值时溢出
        if (N < 0) { // 如果 n < 0,需要将 x 变为其倒数,同时将 n 取相反数
            x = 1 / x;
            N = -N;
        }

        double res = 1.0;
        while (N > 0) {
            if (N % 2 == 1) { // 如果当前 n 是奇数,则将 res 乘上对应的 x 的幂次
                res *= x;
            }
            x *= x; // 更新 x 的幂次
            N /= 2; // 将 n 右移一位,相当于除以 2
        }

        return res;
    }
};

x的平方根

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

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

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

示例 1:

输入:x = 4
输出:2

示例 2:

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

代码

一种朴素的思路是直接遍历[1, x]范围,如果i * i == x就返回i;否则break,返回i - 1

该方法在遍历阶段可以优化,使用二分法遍历效率更高。使用左右指针的中间值来平方,如果该数平方小于x,那么平方获得x的数应该在mid的右边,移动左指针。同理更新右指针,直到找到目标值。

代码如下:

class Solution {
public:
    int mySqrt(int x) {
        if (x == 0) return 0;
        int left = 1;
        int right = x;
        int ans = 0;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (mid <= x / mid) {//实际上就是mid * mid<= x,这么写是为了防止溢出
                ans = mid;
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
        return ans;
    }
};

注意:

1、如果你要显式的使用mid * mid,请将变量定义为long long类型

posted @ 2023-08-06 09:32  dayceng  阅读(42)  评论(0编辑  收藏  举报