编程题练习1-元素查找

来源:互联网 发布:苹果电脑c语言 编辑:程序博客网 时间:2024/06/03 14:48

题目描述:

有一个排过序的数组,包含n个整数,但是这个数组向左进行了一定长度的移位,例如,原数组为[1,2,3,4,5,6],向左移位5个位置即变成了[6,1,2,3,4,5],现在对于移位后的数组,需要查找某个元素的位置。请设计一个复杂度为log级别的算法完成这个任务。

给定一个int数组A,为移位后的数组,同时给定数组大小n和需要查找的元素的值x,请返回x的位置(位置从零开始)。保证数组中元素互异。

测试样例:
[6,1,2,3,4,5],6,6
返回:0


思路:

大家觉得是不是和二分查找特别相像,是的,此题的解法是在二分搜索的基础上加了一些限制条件即可

public int findElement(int[] A, int n, int x) {
        // write code here
        //二分查找
        int left=0;
        int right=n-1;
        int mid=0;
        //由于移位了,但移位之后,中间元素的左右两边必定有一边是升序的  
        while(left<=right) {
            mid=(left+right)/2;
            if(A[mid]==x)
                return mid;
            if(A[mid]<x) {
                //A[mid]<A[left] 说明右边是有序的,且x>A[right]说明x在mid左边
                if(A[mid]<A[left]&&x>A[right]) right=mid-1;
                else left=mid+1;
            }
            else {
                //A[mid]>A[left]说明左边是有序的,且x<A[left],说明x在mid右边
                if(A[mid]>A[left]&&x<A[left]) left=mid+1;
                else right=mid-1;
            }
        }
        return -1;
    }

代码来源自https://www.nowcoder.com/questionTerminal/72ff6503455c4a008675e79247ef2a3a

原创粉丝点击