给定一个有序数组和插入值,使用二分法求出插入位置

9 篇文章 0 订阅

当然对一个有序数组来说,插入一个数,求出插入位置可以使用暴力算法,将插入值与有序数组从头到尾进行比较。这里介绍一种更优的二分法解法,代码如下所示

#include <iostream>
#include <string>
#include <vector>
using namespace std;

int binarySearch(vector<int>& arr, int num, int left, int right) {
    if(left == right) {
        if(num <= arr[left]) {
            return left;
        }
        else {
            return right + 1;
        }
    }

    int mid = (left + right) / 2;
    if(num == arr[mid]) {
        return mid;
    }
    else if(num > arr[mid]) {
        return binarySearch(arr, num, mid + 1, right);
    }
    else {        
        return binarySearch(arr, num, left, mid);
    }
}

int main() {
    vector<int> arr;
    string line;
    string temp;
    while (getline(cin, line)) {
        int i;
        for (i = 0; i < line.size(); i++) {
            if(line[i] == '[') {
                continue;
            }
            if(line[i] == ',') {
                arr.emplace_back(stoi(temp));
                temp.clear();
                continue;
            }
            if(line[i] == ']') {
                arr.emplace_back(stoi(temp));
                temp.clear();
                break;
            }
            temp += line[i];
        }
        int num = stoi(line.substr(i + 2));

        int index = binarySearch(arr, num, 0, arr.size() - 1);
        cout << index << endl;
        arr.clear();
    }

    return 0;
}

在使用二分法求解问题时,需要注意left、right的取值,使用代码中的第21、24行的处理方式能够对最特殊的left、right进行处理,即数组下标为两个相邻正整数(这里以4、5为例),middle=(4+5)/2=4,middle与left相等,在这里,left<=middle<right,所以在将问题一分为二的过程中,有3个分支

分支1:插入数num = 有序数组中间值arr[middle],直接返回middle即可

分支2:插入数num < 有序数组中间值arr[middle],将right更新为middle(切记不是middle-1),继续递归求解

分支3:插入书num > 有序数组中间值arr[middle],将left更新为middle+1(因为middle<right),继续递归求解

递归函数最开始判断left==right,是因为当left与right相等时,插入数num的插入位置必定能够求解出来,无需继续递归

算法的时间复杂度:O(n)

算法的空间复杂度:O(1)

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值