[编程题]二分查找
对于一个有序数组,我们通常采用二分查找的方式来定位某一元素,请编写二分查找的算法,在数组中查找指定元素。
给定一个整数数组A及它的大小n,同时给定要查找的元素val,请返回它在数组中的位置(从0开始),若不存在该元素,返回-1。若该元素出现多次,请返回第一次出现的位置。
测试样例:
[1,3,5,7,9],5,3
返回:1
手撕:
#include<vector>
#include<iostream>
using namespace std;
int getPos(vector<int> A, int n, int val) {
if(A.size()<1)
return -1;
int beg=0;
int end=n-1;
int mid=0;
while(beg<end){
mid=beg+(end-beg)/2;
if(A[mid]>=val){
end=mid;
}else{
beg=mid+1;
}
}
if(A[beg]==val)
return beg;
return -1;
}
测试:
int main(){
//输入1 3 5 7 9
int n=0;
while(cin>>n){
vector<int> vr(n);
for(int i=0;i<n;++i){
cin>>vr[i];
}
int val=0;
cin>>val;
cout<<getPos(vr,n,val)<<endl;
}
return 0;
}
复杂度: log(n)