LeetCode - 33. Search in Rotated Sorted Array

33. Search in Rotated Sorted Array #

Problem's Link
#

 ----------------------------------------------------------------------------

Mean: 

给你一个数组，这个数组是由两个有序的数组拼接成的（无重复元素），在这个数组中查找元素k的下标.

analyse:

既然是由两个有序数组拼接而成，那么只需找到断点，然后进行两次二分查找就行.

Time complexity: O(N)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-03-01-21.22
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long(LL);
typedef unsigned long long(ULL);
const double eps(1e-8);

class Solution
{
public:
   int search(vector<int>& nums, int target)
   {
       int len=nums.size();
       int low1=0,high1=len-1,low2=0,high2=len-1;
       for(int i=1;i<len;++i)
       {
           if(nums[i]<nums[i-1])
           {
               high1=i-1,low2=i;
               break;
           }
       }
       if(high1==len-1 && low2==0)
           return isExist(nums,low1,high2,target);
       else
       {
           int idx1=isExist(nums,low1,high1,target);
           int idx2=isExist(nums,low2,high2,target);
           if(idx1==-1 && idx2==-1)
               return -1;
           else
               return idx1!=-1?idx1:idx2;
       }
   }

   int isExist(vector<int>& nums,int low,int high,int target)
   {
       while(low<=high)
       {
           int mid=(low+high)>>1;
           if(target<nums[mid])
               high=mid-1;
           else if(target>nums[mid])
               low=mid+1;
           else return mid;
       }
       return -1;
   }
};

int main()
{
   Solution solution;
   int n,k;
   vector<int> ve;
   while(cin>>n>>k)
   {
       int tempVal;
       for(int i=0; i<n; ++i)
       {
           cin>>tempVal;
           ve.push_back(tempVal);
       }
       int ans=solution.search(ve,k);
       cout<<ans<<endl;
   }
   return 0;
}
/*

*/