LeetCode题(二分查找,C++实现)
记录一下做题过程,肯定会有比我的更好的实现办法,这里只是一个参考,能帮到大家就再好不过了。
目录
LeetCode题(二分查找,C++实现)
一、搜索插入位置(简单题)
二、在排序数组中查找元素的第一个和最后一个位置(中等题)
三、寻找两个正序数组的中位数(困难题)
一、搜索插入位置(简单题)
题目详情:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n)
的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5 输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2 输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7 输出: 4
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
-
nums
为 无重复元素 的 升序 排列数组 -104 <= target <= 104
解答:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left=0,right=nums.size()-1;
while(left<=right)
{
int mid=left+(right-left)/2;
if(nums[mid]<target)
{
left=mid+1;
}
else
{
right=mid-1;
}
}
return left;
}
};
运行结果:
二、在排序数组中查找元素的第一个和最后一个位置(中等题)
题目详情:
给你一个按照非递减顺序排列的整数数组 nums
,和一个目标值 target
。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target
,返回 [-1, -1]
。
你必须设计并实现时间复杂度为 O(log n)
的算法解决此问题。
示例 1:
输入:nums = [5,7,7,8,8,10], target = 8 输出:[3,4]
示例 2:
输入:nums = [5,7,7,8,8,10], target = 6 输出:[-1,-1]
示例 3:
输入:nums = [], target = 0 输出:[-1,-1]
提示:
0 <= nums.length <= 105
-109 <= nums[i] <= 109
-
nums
是一个非递减数组 -109 <= target <= 109
题解:
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int len=nums.size();
if(len==0)
{
//这里排除数组是空的情况,对应示例3
return vector<int>{-1,-1};
}
//使用自定义函数赋值给firstPosition
int firstPosition=searchFirstPosition(nums,target);
if(firstPosition==-1)
{
//表示找不到第一个与目标数相同的数字,返回[-1,-1]
return vector<int>{-1,-1};
}
//经过排除后,已经没有上面的情况了,肯定能找到第一个数和最后一个数
//找到最后一个数的函数返回值赋值给lastPosition
int lastPosition=searchLastPostion(nums,target);
return vector<int>{firstPosition,lastPosition};
}
int searchFirstPosition(vector<int>& nums,int target)
{
int left=0,right=nums.size()-1;
while(left<right)
{
int mid=left+(right-left)/2;//防止超出时间限制
if(nums[mid]<target)
{
//mid下标比目标元素小,那么mid和mid左边一定不是目标元素第一次出现的位置
//下一轮搜索[mid+1,right]
left=mid+1;
}
else
{
//大于等于目标元素,说明mid或mid左边可能是,目标元素第一次出现的位置
right=mid;
}
}
//当left==right时退出循环,此时left=right=mid;
if(nums[left]==target)
{
//判断此时的left值即此时的mid值是否等于目标数值,等于则返回此时下标
return left;
}
return -1;
}
int searchLastPostion(vector<int>& nums,int target)
{
int left=0,right=nums.size()-1;
while(left<right)
{
int mid=left+(right-left+1)/2;//防止死锁,无法走出循环
if(nums[mid]>target)
{
right=mid-1;
}
else
{
left=mid;
}
}
return left;
}
};
运行结果:
三、寻找两个正序数组的中位数(困难题)
给定两个大小分别为 m
和 n
的正序(从小到大)数组 nums1
和 nums2
。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n))
。
示例 1:
输入:nums1 = [1,3], nums2 = [2] 输出:2.00000 解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4] 输出:2.50000 解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
这一题我也是处于一知半解的状态,放LeetCode官方题解吧
题解:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
if (nums1.size() > nums2.size()) {
return findMedianSortedArrays(nums2, nums1);
}
int m = nums1.size();
int n = nums2.size();
int left = 0, right = m;
// median1:前一部分的最大值
// median2:后一部分的最小值
int median1 = 0, median2 = 0;
while (left <= right) {
// 前一部分包含 nums1[0 .. i-1] 和 nums2[0 .. j-1]
// 后一部分包含 nums1[i .. m-1] 和 nums2[j .. n-1]
int i = (left + right) / 2;
int j = (m + n + 1) / 2 - i;
// nums_im1, nums_i, nums_jm1, nums_j 分别表示 nums1[i-1], nums1[i], nums2[j-1], nums2[j]
int nums_im1 = (i == 0 ? INT_MIN : nums1[i - 1]);
int nums_i = (i == m ? INT_MAX : nums1[i]);
int nums_jm1 = (j == 0 ? INT_MIN : nums2[j - 1]);
int nums_j = (j == n ? INT_MAX : nums2[j]);
if (nums_im1 <= nums_j) {
median1 = max(nums_im1, nums_jm1);
median2 = min(nums_i, nums_j);
left = i + 1;
} else {
right = i - 1;
}
}
return (m + n) % 2 == 0 ? (median1 + median2) / 2.0 : median1;
}
};