PAT-B 1045. 快速排序(25)

题目链接在此。

结合这道题一起看。

思路

思路其实已经在题干中给出了，关键点就在如何表达这个“思路”上。
如果暴力判断的话，肯定是超时的。
将题干思路再深入一层：要判断一个点可不可能是主元，主要判断它左边的数都小于它，它右边的数都大于它。它左边的数小于它深入一层就是它左边最大的数都小于它；同理，右边的数都大于它深入一层就是它右边的最小的数都大于它。

用两个数组leftMax和rightMin，leftMax[i]和rightMin[i]分别表示i左边最大的数和i右边最小的数，这样一来，判断a[i]是不是主元，只需要判断a[i]>leftMax[i] && a[i]

AC代码

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int n;
long long a[100010], ans[100010], num = 0;
//leftMax[i]保存第i位左边的最大值，rightMin[i]保存第i位右边的最小值 
long long leftMax[100010], rightMin[100010];
long long INF = 1<<31-1; //定义成int INF = 1<<31-1 --> 第二个测试点就格式错误 

int main(){

    scanf("%d",&n);
    for(int i = 0; i < n; i++){
        scanf("%lld",&a[i]);
    }

    //填写leftMax
    leftMax[0] = 0;
    for(int i = 1; i < n; i++){
        if(a[i-1] > leftMax[i-1]){
            leftMax[i] = a[i-1];
        }else{ //a[i-1] <= leftMax[i-1] 
            leftMax[i] = leftMax[i-1];
        }
    } 

    //填写rightMin
    rightMin[n-1] = INF;
    for(int i = n-2; i >= 0; i--){
        if(a[i+1] < rightMin[i+1]){
            rightMin[i] = a[i+1];
        }else{ //a[i+1] >= rightMin[i+1] 
            rightMin[i] = rightMin[i+1];    
        }
    } 

    //找可能是主元的数
    for(int i = 0; i < n; i++){
        if(a[i] > leftMax[i] && a[i] < rightMin[i]){ //如果a[i]比它左边最大的数都大，比它右边最小的数都小 
            ans[num++] = a[i];  //则a[i]可能是主元,放入ans数组 
        } 
    } 

    //打印结果 
    printf("%lld\n",num);
    sort(ans,ans+num);
    for(int k = 0; k < num; k++){
        if(k == 0){
            printf("%lld",ans[k]);  
        }else{
            printf("% lld",ans[k]);
        }
    } 
    printf("\n");

    return 0;
}

注：对于INF的数据类型的疑问已经写在备注……问了晴神，等待答复……
有知道的小伙伴现在告诉我真是极好的~