LeetCode(Java)经典题之二分法(34、35、69、367、704)

近期参照《代码随想录》开始刷LeetCode，所以写点笔记来给自己总结，顺便复习。

先从第704题来了解一下二分法吧：

给定一个n个元素有序的（升序）整型数组 nums和一个目标值 target，
写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

首先注意题目要求：有序数组、无重复元素，就立马想到二分法。此处的二分法实现又有两种方法，一种是[left,right]闭区间，一种是[left,right)开区间。根据个人理解方式的不同，在此我选择用第一种方法，采用闭区间时要注意，left<=right的时候也成立，当middle>target的时候，应该让right=middle-1，当middle<target的时候，应该left=middle+1.理清思路之后，实现代码：

public class question704 {
    public int search(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        while (left <= right) {
            int middle = left + (right - left) / 2;
            if (target < nums[middle]) {
                right = middle - 1;
            } else if (target > nums[middle]) {
                left = middle + 1;
            } else {//找到target了，返回数组下标
                return middle;
            }
        }
        return -1;
    }
}

趁热打铁，我们来到第35题

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。
如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

看起来就是在上一题的基础上 ，加上一个插入操作，所以我们只需加一个返回插入的位置即可。

public class question35 {
    public int searchInsert(int[] nums, int target) {
        //闭区间，二分法查找
        int left = 0;
        int right = nums.length - 1;
        while (left <= right) {
            int middle = left + (right - left) / 2;
            if (target < nums[middle]) {
                right = middle - 1;
            } else if (target > nums[middle]) {
                left = middle + 1;
            } else {
                return middle;
            }
        }
        //target在数组所有元素之后，插入到合适的位置
        return right + 1;
    }
}

这两题感觉都不是很复杂（虽然我都是看的答案才做出来），so我们来一道复杂些的，第34题送上。

给定一个按照升序排列的整数数组 nums，和一个目标值 target。
找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target，返回 [-1, -1]。

这一题烦就烦在，数组元素可以有重复的情况，而如果重复之后，需要返回它第一次出现的索引和最后一次出现的索引。我们的二分法一直都只能返回一个索引，所以，需要用两次二分法，一次返回target第一次出现的索引，一次返回target最后一次出现的索引，最后再return出来，即可解决。

public class question34 {
    public int[] searchRange(int[] nums, int target) {
        int leftBorder = getLeftBorder(nums,target);
        int rightBorder = getRightBorder(nums,target);
        // 情况一
        if (leftBorder == -2 || rightBorder == -2) {
            return new int[]{-1, -1};
        }
        // 情况三
        if (rightBorder - leftBorder > 1) {
            return new int[]{leftBorder + 1, rightBorder - 1};
        }
        // 情况二
        return new int[]{-1, -1};
    }

public int getRightBorder(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int rightBorder = -2; // 记录一下rightBorder没有被赋值的情况
        while (left <= right) {
            int middle = left + ((right - left) / 2);
            if (nums[middle] > target) {
                right = middle - 1;
            } else { 
                left = middle + 1;
                rightBorder = left;
            }
        }
        return rightBorder;
    }

public int getLeftBorder(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int leftBorder = -2; // 记录一下leftBorder没有被赋值的情况
        while (left <= right) {
            int middle = left + ((right - left) / 2);
            if (nums[middle] >= target) { 
                right = middle - 1;
                leftBorder = right;
            } else {
                left = middle + 1;
            }
        }
        return leftBorder;
    }
}

不要急，还有两道平方根的题就结束啦。咱们先看第69题：

给你一个非负整数x，计算并返回x的算术平方根 。
 由于返回类型是整数，结果只保留整数部分,小数部分将被舍去 。
 注意：不允许使用任何内置指数函数和算符，例如 pow(x, 0.5)。

 基本思路都与前面的相似，无非就是特殊情况0和1先讨论一下，如果middle的平方大于x了，就把区间缩小到左边，即right=middle-1，反之left=middle+1。具体代码如下：

public class question69 {
    public int mySqrt(int x) {
        if (x == 0) {
            return 0;
        }
        if (x == 1) {
            return 1;
        }
        int left = 1;
        int right = x;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (mid > x / mid) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return left - 1;
    }
}

值得注意的是：一开始我的if判断里面写的是if(mid * mid > x)，结果出现了这样的错误：

猜测可能是因为mid为int类型，范围为-12768~12767，相乘之后溢出，遂改成mid > x/mid，成功。

好的，让我们来看最后一道题，第367题：

给定一个正整数 num,编写一个函数，
如果num是一个完全平方数，则返回true,否则返回 false 。
进阶：不要使用任何内置的库函数，如sqrt 。

这里也就不多强调思路了，相信大家看了代码就会明白：

public class question367 {
    public boolean isPerfectSquare(int num) {
        if(num == 1){
            return true;
        }
        long left = 0;
        long right = num - 1;
        while (left <= right) {
            long middle = left + (right - left) / 2;
            if (middle * middle <= num) {
                left = middle + 1;
            } else {
                right = middle - 1;
            }
        }
        return right * right == num;
    }
}

 前面加了个对于1的判断，是因为一开始没加，提交后报错，所以加上了去（狗头）

 OK，关于二分法就到此结束，希望我三天以后还能记得。