排序数组查找数字
今天实验室同学让我做一道力扣上标记的简单题,题目描述如下:
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
同学跟我说只要我能用低于o(n)时间复杂度的解法做出来就给一瓶牛奶,于是我看完题目后一口气写出如下代码:
class Solution { public: int search(vector<int>& nums, int target) { int left = 0; int right = nums.size(); int mid = 0; int count = 0; while(left < right) { mid = (left + right) / 2; if(nums[mid] > target) { right = mid; } else if(nums[mid] < target) { left = mid+1; } else { count++; break; } } if(count == 0) return 0; for(int i = mid+1; i < nums.size(); i++ ) { if(nums[i] == target) { count++; } else { break; } } for(int i = mid-1; i>=0; i--) { if(nums[i] == target) { count++; } else { break; } } return count; } };
原理如下:先使用二分查找定位目标值,再从目标值左右开弓搜索,是不是有点像中心扩展法
如果按照平均时间复杂度来说,其实还是O(n)级别,并没有比暴力解法好到哪里去,顶多好那么一点点,但他还是送了一瓶牛奶给我,哈哈哈!
下面给出官方解法:
class Solution { public: int binarySearch(vector<int>& nums, int target, bool lower) { int left = 0, right = (int)nums.size() - 1, ans = (int)nums.size(); while (left <= right) { int mid = (left + right) / 2; if (nums[mid] > target || (lower && nums[mid] >= target)) { right = mid - 1; ans = mid; } else { left = mid + 1; } } return ans; } int search(vector<int>& nums, int target) { int leftIdx = binarySearch(nums, target, true); int rightIdx = binarySearch(nums, target, false) - 1; if (leftIdx <= rightIdx && rightIdx < nums.size() && nums[leftIdx] == target && nums[rightIdx] == target) { return rightIdx - leftIdx + 1; } return 0; } }; 作者:LeetCode-Solution 链接:https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof/solution/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-wl6kr/ 来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
原理分析:
通过二分法分别找到左边界和右边界,从而计算出长度,按说这种解法很容易想到,但我当时第一反应还是没想到找左右边界,有时候人的思维就是这么奇妙,明明很简单,但短时间就是不往那个方向去思考,另外,小赌怡情,大赌伤身 0.0
祝各位看官都能发财!