算法:搜索旋转排序数组(二分法延伸)

题目描述

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是 O(log n) 级别。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-in-rotated-sorted-array

分析

数组排序有一下两种情况
图片说明
左图代表完全上升排序未旋转 右图表示旋转过
所以算法框架还是二分法查找,只是需要分情况

class Solution {
    public int search(int[] nums, int target) {
        if(nums == null || nums.length == 0) return -1;
        int l = 0;
        int r = nums.length-1;
        while(l <= r) {
            int mid = (l+r)/2;
            if(nums[mid] == target)  //找到目标
                return mid;
            else if(nums[mid] < nums[r]) {   //左图情况
                if(nums[mid] < target && target <= nums[r]) {  //目标在右上段,二分法向右移动
                    l = mid + 1;
                }
                else {      //目标在左下段,二分法向左移动
                    r = mid - 1;
                }                
            }
            else {  //右图情况
                if(nums[l] <= target && target < nums[mid]) { //目标在左上段,二分法向左移动
                    r = mid - 1;
                }
                else {     //目标在右下段,二分法向右移动
                    l = mid + 1;
                }
            }

        }
        return -1;
    }

}
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务