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