首页 > 其他 > 详细

LeetCode - 33. Search in Rotated Sorted Array

时间:2016-03-01 22:45:41      阅读:252      评论:0      收藏:0      [点我收藏+]

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;
}
/*

*/

 

LeetCode - 33. Search in Rotated Sorted Array

原文:http://www.cnblogs.com/crazyacking/p/5232699.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!