欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

【二分查找】35. 搜索插入位置

程序员文章站 2022-10-06 08:22:55
利用二分查找算法和C++短路求值特性。 平台AC代码: 1 class Solution { 2 public: 3 int searchInsert(vector& nums, int target) { 4 int lo = 0, hi = nums.size(); 5 while ( ......

利用二分查找算法c++短路求值特性

平台ac代码:

 1 class solution {
 2 public:
 3     int searchinsert(vector<int>& nums, int target) {
 4         int lo = 0, hi = nums.size();
 5         while (lo < hi){
 6             int mi = lo + ((hi - lo) >> 1);
 7             (target < nums[mi]) ? hi = mi : lo = mi + 1;
 8         }
 9         int i = --lo;
10         if (i == -1 || nums[i] != target) i++;
11         return i;
12     }
13 };

c++自测代码:

 1 #include <iostream>
 2 
 3 using namespace std;
 4 
 5 int binarysearch(int* a, int e, int lo, int hi)
 6 {
 7     while (lo < hi){
 8         int mi = lo + ((hi - lo) >> 1);  //避免数据溢出
 9         (e < a[mi]) ? hi = mi : lo = mi + 1;
10     }
11     return --lo;
12 }
13 
14 int main()
15 {
16     int a[] = {1,3,5,6};
17     int target;
18     while(cin >> target){
19         int i = binarysearch(a, target, 0, 4);
20         if (i == -1 || a[i] != target) i++;  // 利用c++短路求值特性,避免数组越界情况。
21         cout << i << endl;
22     }
23     return 0;
24 }