天天看點

搜尋旋轉排序數組(c++)

  1. 題目:

    整數數組 nums 按升序排列,數組中的值 互不相同 。

在傳遞給函數之前,nums 在預先未知的某個下标 k(0 <= k < nums.length)上進行了 旋轉,使數組變為 [nums[k], nums[k+1], …, nums[n-1], nums[0], nums[1], …, nums[k-1]](下标 從 0 開始 計數)。例如, [0,1,2,4,5,6,7] 在下标 3 處經旋轉後可能變為 [4,5,6,7,0,1,2] 。

給你 旋轉後 的數組 nums 和一個整數 target ,如果 nums 中存在這個目标值 target ,則傳回它的下标,否則傳回 -1 。

來源:力扣(LeetCode)

連結:https://leetcode-cn.com/problems/search-in-rotated-sorted-array

著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。

2. 解題思路:

這題想要時間複雜度控制再O(log n)我首先想到的就是采用折半查找的方法,但是這題的數組是事先被旋轉過的,自然而然先想到的就是把這個k的值找出來,然後再進行分段的折半查找,這題要注意邊界值的處理。

3. 解題代碼:

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        int k = 0;
        int mid;
        //找k
        while(k+1 < n){
            if(nums[k] < nums[k+1]){
                k++;
            }else{
                break;
            }
        }
        int left = 0 , right = n-1;
		//當長度大于1時進行left和right 的設定
        if(n > 1){
            if(nums[left] < target){
                left = 1;
                right = k;
            }
            else if(nums[left] == target){
                return left;
            }
            else if(nums[n-1] > target){
                left = k+1;
                right = n-2;
            }
            else if(nums[n-1] == target){
                return n-1;
            }
        }else{ //當長度為1 的時候隻用判斷nums[0]的值是否是target
            if(nums[0] == target){
                return 0;
            }else{
                return -1;
            }
        }
        
        //進行折半查找
        while(left <= right){
            mid = (left+right)/2;
            if(nums[mid] == target){
                return mid;
            } 
            if(nums[mid] > target){
                right = mid-1;
            }
            if(nums[mid] < target){
                left = mid +1;
            }
        }

        return -1;
    }
};
           
  1. tips:

    折半查找很适合對基本有序的數組進行查找,效率很高。