[二分查找] 旋转数组

1. （严格递增序列）旋转数组的元素查找

 简单来说分为三种情况进行分析

1. 整个旋转数组单调递增

根据x和A[mid]的大小关系，更迭范围。

        // 1. 整个旋转数组单调递增
        if (A[left]<A[right]){
            if (A[mid] == x)return mid;
            else if (x < A[mid])right = mid-1;
            else if (x > A[mid])left = mid+1;
        }

2. 旋转数组分为两个递增序列，mid位于第一个递增序列当中

mid将旋转数组分为两块区间，但第一块区间的值的条件并不是简单的x<A[mid]，因为小于A[mid]的元素还有可能为最下面的区间，如图。

        // 2. 旋转数组分为两个递增序列，mid位于第一个递增序列当中
        if (A[mid]>A[left]){
            if (x == A[mid])return mid;
            else if (x<A[mid] && x >= A[left]){
                right = mid-1;
            }else{
                left = mid+1;
            }
        }

 3. mid位于第二个递增序列当中

        // 3. 旋转数组分为两个递增序列，mid位于第二个递增序列当中
        if (A[mid]<A[left]){
            if (x == A[mid])return mid;
            else if (x>A[mid] && x<=A[right]){
                left = mid+1;
            }else {
                right = mid-1; 
            }
        }

最后的总代码：

#include <cstdio>

const int N= 1e5+10;
int A[N];
int n,x;
int binary_search(int A[],int left,int right,int x){
    int mid;
    while (left < right){
        mid = left + (right-left)/2;
        
        // 1. 整个旋转数组单调递增
        if (A[left]<A[right]){
            if (A[mid] == x)return mid;
            else if (x < A[mid])right = mid-1;
            else if (x > A[mid])left = mid+1;
        }
        // 2. 旋转数组分为两个递增序列，mid位于第一个递增序列当中
        if (A[mid]>A[left]){
            if (x == A[mid])return mid;
            else if (x<A[mid] && x >= A[left]){
                right = mid-1;
            }else{
                left = mid+1;
            }
        }
        // 3. 旋转数组分为两个递增序列，mid位于第二个递增序列当中
        if (A[mid]<A[left]){
            if (x == A[mid])return mid;
            else if (x>A[mid] && x<=A[right]){
                left = mid+1;
            }else {
                right = mid-1; 
            }
        }
    }
    if (A[left]==x)return left;
    return -1;
}
int main(){

    scanf("%d%d",&n,&x);
    for (int i=0;i<n;i++)scanf("%d",&A[i]);

    printf("%d",binary_search(A,0,n-1,x));
    return 0;

}

2.（非递减序列）旋转数组的元素查找

思路一致，由于会出现重复的元素，因此，对寻找第一个x的寻找条件进行修改即可。

不过由于非递减的条件不好写，因此我们可以用else来表示

#include <cstdio>

const int N= 1e5+10;
int A[N];
int n,x;
int binary_search(int A[],int left,int right,int x){
    int mid;
    while (left < right){
        mid = left + (right-left)/2;

        // 2. 旋转数组分为两个非递减序列，mid位于第一个非递减序列当中
        if (A[mid]>A[left]){
            if (x == A[mid])right = mid;
            else if (x<A[mid] && x >= A[left]){
                right = mid-1;
            }else{
                left = mid+1;
            }
        }else if (A[mid]<A[left]){
            // 3. 旋转数组分为两个递增序列，mid位于第二个递增序列当中
            if (x == A[mid])right=mid;
            else if (x>A[mid] && x<=A[right]){
                left = mid+1;
            }else {
                right = mid-1;
            }
        }else {
            //1. 整个旋转数组非递减
            if (A[mid] == x)right = mid;
            else if (x < A[mid])right = mid-1;
            else if (x > A[mid])left = mid+1;
        }
    }
    if (A[left]==x)return left;
    return -1;
}
int main(){

    scanf("%d%d",&n,&x);
    for (int i=0;i<n;i++)scanf("%d",&A[i]);

    printf("%d",binary_search(A,0,n-1,x));
    return 0;

}

3. 旋转数组的最小值，通过数组左端元素判断最小值出现的位置

两个代码相同。需要注意 A[left] == A[mid] 的情况，最小值在右侧区间中

#include <cstdio>

const int N= 1e5+10;
int A[N];
int n;
int binary_search(int A[],int left,int right){
    int mid;
    while (left < right){
        mid = left + (right-left)/2;
        // 1. 整个旋转数组单调递增
        if (A[left]<A[right]){
            return A[left];
        }
        // 2. 旋转数组分为两个单调递增序列，mid位于第一个单调递增序列当中
        if (A[mid]>A[left]){
            left = mid+1;
        }else if (A[mid]<A[left]){
            // 3. 旋转数组分为两个递增序列，mid位于第二个递增序列当中
            right = mid;
        }else if(A[mid]==A[left]){
            // A[left] == A[mid] 的情况，最小值在右侧区间中
            left = mid+1;
        }
        //printf("%d,%d\n",left,right);
    }
    return A[left];
}
int main(){

    scanf("%d",&n);
    for (int i=0;i<n;i++)scanf("%d",&A[i]);

    printf("%d",binary_search(A,0,n-1));
    return 0;

}

4. 旋转数组的中位数

#include <cstdio>

const int N= 1e5+10;
int A[N];
int n;
int binary_search(int A[],int left,int right){
    int mid;
    while (left < right){
        mid = left + (right-left)/2;
        // 1. 整个旋转数组单调递增
        if (A[left]<A[right]){
            return left;
        }
        // 2. 旋转数组分为两个单调递增序列，mid位于第一个单调递增序列当中
        if (A[mid]>A[left]){
            left = mid+1;
        }else if (A[mid]<A[left]){
            // 3. 旋转数组分为两个递增序列，mid位于第二个递增序列当中
            right = mid;
        }else if(A[mid]==A[left]){
            // A[left] == A[mid] 的情况，最小值在右侧区间中
            left = mid+1;
        }
        //printf("%d,%d\n",left,right);
    }
    return left;
}
int main(){

    scanf("%d",&n);
    for (int i=0;i<n;i++)scanf("%d",&A[i]);

    int minpos = binary_search(A,0,n-1);
    //printf("%d",minpos);
    if (n%2==0){
        printf("%.1f",(A[(minpos+n/2)%n]+A[(minpos+n/2)%n-1])/2.0);
    }else {
        printf("%.1f",A[(minpos+n/2)%n]/1.0);
    }
    return 0;

}

 重点在于后面中位数的求解

    if (n%2==0){
        printf("%.1f",(A[(minpos+n/2)%n]+A[(minpos+n/2)%n-1])/2.0);
    }else {
        printf("%.1f",A[(minpos+n/2)%n]/1.0);
    }