二分查找并插入java实现

记录第二道想记录的二分算法题

这个题是二分查找的第三题，个人觉得也有必要纪念一下。首先先看本道题的题目

本道题不仅需要查找是否有相同的变量，还需要假如未找到相同的变量，还需要将所要查询的变量插入到数组中。在刚开始本人只是查找是否相同，然后就想当然的认为，假如找不到相同的变量，那么mid值就是需要插入的下标值，但是再提交了几次以后，发现不太对劲，后续又想到了在相等情况下假如还未找到相等的数值之后，mid的值会进行一次新的改变，这个时候就需要判断mid值是比target值大还是小，假如比target值大的话，那么就需要将mid+1的值给返回出去，假如比target小的话，那么就需要将mid的值返回出去
代码如下

class Solution {
    public int searchInsert(int[] nums, int target) {
        int a = 0,b = nums.length-1;
        int mid=0;
        while(a<=b){
            mid = a + (b - a)/2;
            if(nums[mid]==target){
                return mid;
            }
            if(nums[mid]>target){
                b = mid-1;
            }
            if(nums[mid]<target){
                a = mid+1;
            }
        }
        if(nums[mid]>target){
            return mid;
        }
        else{
            return mid+1;
        }
    }
}