【leetcode】35. 搜索插入位置，69.X的平方根，278. 第一个错误的版本

35. {二分查找}搜索插入位置,69.x的平方根，278. 第一个错误的版本

题目描述

题目分析

首先一方面可以暴力直接单指针挨个比，比到合适的就插进去，复杂度O(N)

但是因为是个有序数组，可以使用二分法，顺便学习了一下二分法

题解代码

挨个比

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
            int index = 0;
            if(target>nums[nums.size()-1])
        return nums.size();
    for(int i = 0;i < nums.size();i++)
        if(target <= nums[i])
        {
            index = i;
            break;
        }
        return index;
    }
};

二分法

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
    if(nums.size()==0) return 0;
    if(nums[nums.size()-1]<target) return nums.size(); 
    int left = 0,right = nums.size()-1;
    int mid;
    while(left<right)
    {
        mid  = left + (right - left)/2;
        if(nums[mid]<target)
            left = mid + 1;
        else
            right = mid ;
    } 
    return left;
    }
};

69.X的平方根

题目描述

题目分析

同样位二分法查找，不多赘述

题解代码

class Solution {
public:
    int mySqrt(int x) {
            long long  left = 0 ,right = x,mid;
         while(left<right)
    {
        mid = left + (right - left + 1)/2;
        if(mid*mid <= x)
            left = mid;
        else
            right = mid-1;
    }
        return left;
    }
};

278. 第一个错误的版本

题目描述

题目分析

同样位二分法查找，不多赘述

题解代码

bool isBadVersion(int version);
class Solution {
public:
    int firstBadVersion(int n) {
    int l = 1,r=n;
    while(l<r)
    {
        int mid = l + (r - l)/2;
        if(!isBadVersion(mid))
            l = mid + 1 ;
        else
            r = mid;
    }
    return l;
    }
};

【模板】二分查找模板（转）

二分查找链接模板原文链接

1.求最小的i，使得a[i] = key，若不存在，则返回-1

int binary_search_1(int a[], int n, int key)
{
    int m, l = 0, r = n - 1; // 闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r - l) >> 1); // 向下取整
        if (a[m] < key) l = m + 1;
        else r = m;
    }
    if (a[r] == key) return r;
    return -1;
}

2.求最大的i，使得a[i] = key，若不存在，则返回-1

int binary_search_2(int a[], int n, int key)
{
    int m, l = 0, r = n - 1; // 闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r + 1 - l) >> 1); // 向上取整
        if (a[m] <= key) l = m;
        else r = m - 1;
    }
    if (a[l] == key) return l;
    return -1;
}

3.求最小的i，使得a[i] > key，若不存在，则返回-1

int binary_search_3(int a[], int n, int key)
{
    int m, l = 0, r = n - 1;//闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r - l) >> 1);//向下取整
        if (a[m] <= key) l = m + 1;
        else r = m;
    }
    if (a[r] > key) return r;
    return -1;
}

4.求最大的i，使得a[i] < key，若不存在，则返回-1

int binary_search_4(int a[], int n, int key)
{
    int m, l = 0, r = n - 1;//闭区间[0, n - 1]
    while (l < r)
    {
        m = l + ((r + 1 - l) >> 1);//向上取整
        if (a[m] < key) l = m;
        else r = m - 1;
    }
    if (a[l] < key) return l;
    return -1;
}

实际上，对于3、4，也可以先判断解是否存在，再进行二分查找。
显然，上面的代码还不够简洁。下面给出我认为最简洁的代码：

// 1.
int ans = std::lower_bound(a, a + n, key) - a;
ans = (ans == n || a[ans] != key) ? -1 : ans;
 
// 2.
int ans = std::upper_bound(a, a + n, key) - a;
ans = (ans == 0 || a[ans - 1] != key) ? -1 : ans - 1;
 
// 3.
int ans = std::upper_bound(a, a + n, key) - a;
ans = (ans == n) ? -1 : ans;
 
// 4.
int ans = std::lower_bound(a, a + n, key) - a;
ans = (ans == 0) ? -1 : ans - 1;