改写二分搜索算法 思路与分析
题目来源:《计算机算法设计与分析》,王晓东
设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
——————————————————————————————————分割线——————————————————————————————————————————————
思路:对于能够找到的元素,只需要把这个下标再输出一遍即可。
对于找不到的元素,我们从优化过的二分搜索算法着手。众所周知,二分搜索算法的左指针和右指针会渐渐靠近,如果要输出这个数离得最近的两个下标,不妨分析left和right的接近情况。【这里我用的是left>right的递归条件,这样子做除了简化代码,还有特殊用处】
知识兔括号左右是每次递归传递给形参的left right的值。
找0:(0,3)->(0,0)->(0,-1)
找4:(2,3)->(3,3)->(3,2)
找6:(2,3)->(3,3)->(4,3)
【记住:mid=(l+r)/2会取整,mid每次都要+1或者-1】
可以发现,只要把返回-1前的左右两个值调换顺序,就是正确的输出结果。这里我选择用全局变量保存。
然后在主函数里面判断是不是返回-1,以选择不同的输出方式。
时间复杂度:O(logn)【二分搜索,不解释】空间复杂度O(n)
收获:当你看到大概知道思路但是解不开的题目,在大脑里面模拟运行,可能就有意外收获。
最后附上代码
1 #include<iostream>
2 using namespace std;
3 int leftnum=0;
4 int rightnum=0;
5 int binarysearch(int num[],int left,int right,int chazhao)
6 {
7 //findnum++;
8 int mid;
9 if(left>right)
10 {
11 //cout<<"zhaobudao";
12 leftnum = left;
13 rightnum = right;
14 return -1;
15 }
16 else
17 {
18 mid=(left+right)/2;
19 if(num[mid]==chazhao)
20 {
21 return mid;
22 }
23 else if(num[mid]!=chazhao)
24 {
25 if(chazhao>num[mid])
26 {
27 return binarysearch(num,mid+1,right,chazhao);
28 }
29 else return binarysearch(num,left,mid-1,chazhao);
30 }
31 }
32 }
33
34
35 int main()
36 {
37 int num[1000];
38 int i,chazhao,n;
39 cin>>n;
40 cin>>chazhao;
41 for(i=0;i<n;i++)
42 {
43 cin>>num[i];
44 }
45
46 int date = binarysearch(num,0,n-1,chazhao);
47
48 if(date == -1){
49 cout<<rightnum<<" "<<leftnum;
50
51 }
52 else
53 {
54 cout<<date<<" "<<date;
55 }
56
57 }
知识兔
知识兔