DS二分查找_搜索旋转排序数组

题目描述

长度为n的整型数组nums原本是按照升序排列的，接着在某个未知的下标k(0 <= k < n)处进行了旋转，使得数组变为[nums[k], nums[k + 1], ... , nums[n - 1], nums[0], nums[1], ... , nums[k - 1]]。例如，[0,1,2,3,4,5,6,7]在下标3处进行旋转后变为[3,4,5,6,7,0,1,2]。注意k可以是0。

给定旋转后的数组nums和一个整数target，如果nums中存在这个目标值target，则输出它的下标，否则输出-1。

要求使用二分查找法解题。

输入

第一行输入t，表示有t个测试样例。

对于每个测试样例，首先输入n，接着输入n个整数，接着输入target。

以此类推，共输入t个测试样例。

5

1
1
0

2
1 2
2

3
2 3 1
5

4
4 1 2 3
4

7
4 5 6 7 0 1 2
3

输出

每一行输出查找结果，若nums中存在这个目标值target，则输出它的下标，否则输出-1。

共输出t行。

-1
1
-1
0
-1

提示

1 <= n <= 5000

-10^4 <= nums[i] <= 10^4

-10^4 <= target <= 10^4

nums中的每个值都独一无二。

题目数据保证nums在预先未知的某个下标上进行了旋转。

#include <iostream>
using namespace std;
int search(int* a, int target, int n)
{
    int max = 0, min, mid;
    for (int i = 0; i < n; i++)//找出发生旋转的点
    {
        if (a[0] > a[i])
        {
            max = i;
            break;
        }
    }
    if (max == 0)//当旋转发生在0处时，就是没有发生旋转
        max = n - 1;
    else if (target<a[max] || target>a[max - 1])//目标超出数组的大小返回-1
        return -1;
    if (target >= a[0])//以旋转的地方为中心，将数组分为左半边和右半边，每个半区的数组是递增的，可以使用二分法查找
    {//左半区
        min = 0;
        while (min <= max)
        {
            mid = (min + max) / 2;
            if (a[mid] == target)
                return mid;
            else if (a[mid] > target)
                max = mid - 1;
            else if (a[mid] < target)
                min = mid + 1;
        }
        return -1;
    }
    else//右半区
    {
        min = max;
        max = n - 1;
        while (min <= max)
        {
            mid = (min + max) / 2;
            if (a[mid] == target)
                return mid;
            else if (a[mid] > target)
                max = mid - 1;
            else if (a[mid] < target)
                min = mid + 1;
        }
        return -1;
    }
}

int main()
{
    int t, n;

    cin >> t;
    while (t--)
    {
        cin >> n;
        int* a = new int[n];
        for (int i = 0; i < n; i++)
        {
            cin >> a[i];//录入数组
        }
        int target;
        cin >> target;
        cout << search(a, target, n) << endl;
        delete[]a;
    }
}