代码随想录Day1

二分查找

最开始AC的代码,没有考虑到数组是有序递增的,可以先判断一下target是否小于nums[0]或者大于nums[r],这里算是一个点可以优化。

class Solution {
    public int search(int[] nums, int target) {
        int l = 0, r = nums.length - 1;
        while (l <= r) {
            int mid = (l + r) >> 1;
            if(nums[mid] > target) {
                r = mid - 1;
            } else if (nums[mid] < target) {
                l = mid + 1;
            } else {
                return mid;
            }
        }
        return -1;
    }
}

优化之后的代码

class Solution {
    public int search(int[] nums, int target) {
        int l = 0, r = nums.length - 1;
        if (target < nums[l] || target > nums[r]) {
            return -1;
        }

        while(l <= r) {
            int mid = (l + r) >> 1;
            if(nums[mid] > target) {
                r = mid - 1;
            } else if (nums[mid] < target) {
                l = mid + 1;
            } else {
                return mid;
            }
        }
        return -1;
    }
}

同时明确了二分法对于区间边界不同,写法也不同。

移除元素

之前学习过这个方法

class Solution {
    public int removeElement(int[] nums, int val) {
        int slow = 0;
        for (int fast = 0; fast < nums.length; fast++) {
            if (nums[fast] != val) {
                nums[slow] = nums[fast];
                slow++;
            }
        }   
        return slow;
    }
}

你可能感兴趣的:(算法刷题,leetcode)