时间复杂度:O(N) 空间复杂度:O(1)
public int search(int[] nums, int target) { int count = 0; for(int num : nums){ if(num == target){ count++; } } return count; }有序数组 分别找到 taret 的左右边界 right - left +1 时间复杂度:O(logN) 空间复杂度:O(1)
public class Solution { public int search(int[] nums, int target) { int len = nums.length; if (len == 0) { return 0; } int firstPosition = findFirstPosition(nums, target); if (firstPosition == -1) { return 0; } int lastPosition = findLastPosition(nums, target); return lastPosition - firstPosition + 1; } private int findFirstPosition(int[] nums, int target) { int len = nums.length; int left = 0; int right = len - 1; while (left < right) { int mid = left + (right - left) / 2; // 注意这样写,可以从左边收缩待搜索区间的范围,进而找到第一次出现的位置 if (nums[mid] < target) { // mid 以及 mid 左边都不是,下一轮搜索区间在 [mid + 1, right] left = mid + 1; } else { right = mid; } } if (nums[left] == target) { return left; } return -1; } private int findLastPosition(int[] nums, int target) { int len = nums.length; int left = 0; int right = len - 1; while (left < right) { int mid = left + (right - left + 1) / 2; // 注意这样写,可以从右边收缩待搜索区间的范围,进而找到最后一次出现的位置 if (nums[mid] > target) { // mid 以及 mid 右边都不是,下一轮搜索区间在 [left, mid - 1] right = mid - 1; } else { left = mid; } } return left; } }1、确定搜索区间初始化时候的左右边界,有时需要关注一下边界值。在初始化时,有时把搜索区间设置大一点没有关系,但是如果恰好把边界值排除在外,再怎么搜索都得不到结果。
2、无条件写上 while (left < right) ,表示退出循环的条件是 left == right,对于返回左右边界就不用思考了,因此此时它们的值相等;
3、先写下取整的中间数取法,然后从如何把 mid 排除掉的角度思考 if 和 else 语句应该怎样写。
(这里建议写两个注释。)
一般而言,我都会把**“什么时候不是目标元素”**作为注释写在代码中,提醒自己要判断正确,这一步判断非常关键,直接影响到后面的代码逻辑。然后接着思考 mid 不是解的情况下,mid 的左右两边可能存在解,把下一轮搜索的区间范围作为注释写进代码里,进而在确定下一轮搜索区间边界的收缩行为时,不容易出错。if 有把握写对的情况下,else 就是 if 的反面,可以不用思考,直接写出来。
** 说明:这种思考方式,就正正好把待搜索区间从逻辑上分成两个区间,一个区间不可能存在目标元素,进而在另一个区间里继续搜索,更符合“二分”的语义。**
4、根据 if else 里面写的情况,看看是否需要修改中间数下取整的行为。 上面已经说了,只有看到 left = mid 的时候,才需要调整成为上取整,记住这一点即可,我因为刚开始不理解这种写法,遇到很多次死循环,现在已经牢记在心了。
5、退出循环的时候,一定有 left == right 成立。有些时候可以直接返回 left (或者 right,由于它们相等,后面都省略括弧)或者与 left 相关的数值,有些时候还须要再做一次判断,判断 left 与 right 是否是我们需要查找的元素,这一步叫“后处理”。
// 有可能区间内不存在目标元素,因此还需做一次判断 if (nums[left] == target) { return left; } return -1; public int searchInsert(int[] nums, int target) { int len = nums.length; if (len == 0) { return 0; } int left = 0; // 因为有可能数组的最后一个元素的位置的下一个是我们要找的,故右边界是 len int right = len; while (left < right) { int mid = (left + right) >>> 1; // 小于 target 的元素一定不是解 if (nums[mid] < target) { // 下一轮搜索的区间是 [mid + 1, right] left = mid + 1; } else { right = mid; } } return left; }
转载链接:https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof/solution/er-fen-cha-zhao-fa-zhao-dao-di-yi-ci-chu-xian-de-w/