这几天一直在PTA上写算法题,二分查找花了我挺长时间,在博客记录一下。
一、二分查找 (20分)
题目
输入n值(1<=n<=1000)、n个非降序排列的整数以及要查找的数x,使用二分查找算法查找x,输出x所在的下标(0~n-1)及比较次数。若x不存在,输出-1和比较次数。
输入格式
输入共三行: 第一行是n值; 第二行是n个整数; 第三行是x值。
输出格式
输出x所在的下标(0~n-1)及比较次数。若x不存在,输出-1和比较次数。
输入样例
4
1 2 3 4
1
输出样例
0
2
一直写却总是有一个或两个错误,我就去百度了一个二分查找,但是这个不是基于递归的。
非递归的二分查找(java)
import java.util.*;
public class test {
static int count=0;//比较次数
static int binarySearch(int[] a,int x,int low,int high){
while(high>=low){//注意直接用while就行,递归栈溢出
count++;
int middle=(low+high)/2;
if(a[middle]==x){
return middle;
}
else if(a[middle]>x){
high=middle-1;
}
else{
low=middle+1;
}
}
return -1;
}
public static void main(String[] args) {
// TODO 自动生成的方法存根
Scanner in=new Scanner(System.in);
int n=in.nextInt();
int[] a=new int[n];
for(int i=0;i<n;i++){
a[i]=in.nextInt();
}
int x=in.nextInt();
System.out.println(binarySearch(a,x,0,a.length-1));//下标);
System.out.println(count);
}
}
看到这个程序,我似乎理解了我写的二分算法是不对的,二分算法其实是将要寻找的数字与目前头,尾的中间的数作比较,即与middle比较,根据比较结果移动初始位置或结尾位置。而且数组下标移动应为(start+end)/2,而不是(end - start)/2,修改如下:
以下算法是通过PTA基于递归的二分查找
//递归算法,前面提交虽然正确,但不是递归,之前一直错在(start-end)/2的地方,应该是(start+end)/2!!!
//对于二分查找理解有误,二分查找必须是一个按照大小关系排列好的序列来使用
#include<iostream>
using namespace std;
#define N 1000
int erfen(int start,int end,int array[],int find,int *count){
(*count)++;
if(array[(start + end) / 2] == find){
return (start + end) / 2;
}
else if(start == end && array[(start + end) / 2] != find){
return -1;
}
else if(array[(start + end) / 2] > find){
if((start + end) / 2 - 1 == -1)
return -1;
return erfen(start,(start + end) / 2 - 1,array,find,count);
}
else if(array[(start + end) / 2] < find){
return erfen((start + end) / 2 + 1,end,array,find,count);
}
return -1;
}
int main(){
int n,i,find,count = 0; //n为数字个数,find为要查找的数,count统计次数
cin>>n;
if(n>1000 || n<1) return 0;
int array[N];
for(i = 0;i < n;i++)
cin>>array[i];
cin>>find;
int index = erfen(0,n-1,array,find,&count);
cout<<index<<endl;
cout<<count<<endl;
return 0;
}
二、改写二分搜索算法 (20分)
题目来源:《计算机算法设计与分析》,王晓东
题目
设a[0:n-1]是已排好序的数组,请改写二分搜索算法,使得当x不在数组中时,返回小于x的最大元素位置i和大于x的最小元素位置j。当搜索元素在数组中时,i和j相同,均为x在数组中的位置。
输入格式
输入有两行
第一行是n值和x值; 第二行是n个不相同的整数组成的非降序序列,每个整数之间以空格分隔。
输出格式
输出小于x的最大元素的最大下标i和大于x的最小元素的最小下标j。当搜索元素在数组中时,i和j相同。 提示:若x小于全部数值,则输出:-1 0
若x大于全部数值,则输出:n-1的值 n的值
输入样例
在这里给出一组输入。例如:
6 5
2 4 6 8 10 12
输出样例
在这里给出相应的输出。例如:
1 2
这道题就是在上面那道题的基础上,在不同情况下进行赋值
#include<iostream>
using namespace std;
#define N 100000
int erfen(int start,int end,int array[],int find,int *i,int *j){
if(array[(start + end) / 2] == find){//如果当前位置的数就是要找的数
(*i) = (start + end) / 2; //给i指向的空间赋值成当前位置的值
(*j) = (*i); //给j指向的空间赋值成当前位置的值
return 0;
}
else if(start == end && array[(start + end) / 2] != find){//临界条件,说明没有找到这个数
if(start == 0){ //这是要寻找的数小于当前所有的数组数情况
(*i) = -1;
(*j) = 0;
}
else{ //其他情况下,这种计算方式通用
(*i) = (start + end) / 2;
(*j) = (start + end) / 2 + 1;
}
return 0;
}
else if(array[(start + end) / 2] > find){ //如果当前数组中间的数大于寻找的数
if((start + end) / 2 - 1 == -1){
(*i) = -1;
(*j) = 0;
return 0;
}
erfen(start,(start + end) / 2 - 1,array,find,i,j); //对数组的左边再进行二分查找
return 0;
}
else if(array[(start + end) / 2] < find){ //如果当前数组中间的数小于寻找的数
erfen((start + end) / 2 + 1,end,array,find,i,j); //对数组的右边再进行二分查找
return 0;
}
return 0;
}
int main(){
int n,i,j = 0,find; //n为数字个数,find为要查找的数,count统计次数
cin>>n;
cin>>find;
int array[N];
for(i = 0;i < n;i++)//向数组中输入数字
cin>>array[i];
erfen(0,n-1,array,find,&i,&j);
cout<<i<<" "<<j<<endl;
return 0;
}