[每日一题]40:元素查找

122 篇文章 36 订阅
51 篇文章 0 订阅

题目描述

有一个排过序的数组,包含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

思路:

在二分查找的基础上,稍作改变即可。

具体方法如下: 在解本次前一定要抓住:数组是经过移位的,那么中间元素的两侧肯定有一边是升序的。 找到数组A中间位置mid,确定目标数 x 在该数的哪一边。

  1. 如果x == A[mid]时, 找到了返回mid

  2. 当x大于A[mid]时,分两种情况:
    (1)如果数组开头数A[left]大于A[mid],说明右半边是升序,因为移位之后,肯定会将大的元素移到左边,此时,如果x > A[right]时,说明x在mid的左半侧
    (2)否则:x位于区间右半侧

  3. 当x小于mid,分两种情况:
    (1)如果数组开头元素A[left] < A[mid], 说明左边是有序的,此时如果x < A[left], 说明x在区间的右半侧
    (2)否则:x位于区间左半侧

代码解答:

class Finder {
public:
    int findElement(vector<int> A, int n, int x) {
        int left = 0;
        int right = n - 1;
        int mid;
        
        while(left <= right){
            mid = left + (right - left) / 2;
            
            if(A[mid] == x){
                return mid;
            }
            
            if(x > A[mid]){
                //A[mid] < A[left] 说明右边是有序的,且 x > A[right]说明x在mid左边
                if(A[left] > A[mid] && A[right] < x){
                    right = mid - 1;
                }
                else{
                    left = mid + 1;
                }
            }
            else{
                //A[mid]>A[left]说明左边是有序的,且x<A[left],说明x在mid右边
                if(A[left] < A[mid] && A[left] > x){
                    left = mid + 1;
                }
                else{
                    right = mid - 1;
                }
            }
        }
        return -1;
    }
};

如有不同见解,欢迎留言讨论!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值