1045. 快速排序(25)

1045. 快速排序(25)

时间限制 200 ms
内存限制 65536 kB
代码长度限制 8000 B
判题程序 Standard
作者 CAO, Peng

著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。 给定划分后的N个互不相同的正整数的排列，请问有多少个元素可能是划分前选取的主元？

例如给定N = 5, 排列是1、3、2、4、5。则：

1的左边没有元素，右边的元素都比它大，所以它可能是主元；
尽管3的左边元素都比它小，但是它右边的2它小，所以它不能是主元；
尽管2的右边元素都比它大，但其左边的3比它大，所以它不能是主元；
类似原因，4和5都可能是主元。
因此，有3个元素可能是主元。

输入格式：

输入在第1行中给出一个正整数N（<= 10^5）； 第2行是空格分隔的N个不同的正整数，每个数不超过109。

输出格式：

在第1行中输出有可能是主元的元素个数；在第2行中按递增顺序输出这些元素，其间以1个空格分隔，行末不得有多余空格。

输入样例：
5
1 3 2 4 5
输出样例：
3
1 4 5

原题地址： https://www.patest.cn/contests/pat-b-practise/1045

---

第i个元素a[i]如果是主元应该满足：
max(a[0], a[1], …, a[i-1]) <= a[i] <= min(a[i + 1], a[i + 2], …, a[N - 1])
记
leftMax[i] = max(a[0], a[1], …, a[i - 1])
rightMin[i] = min(a[i + 1], a[i + 2], …, a[N - 1])
所以我们只要预先算出leftMax和rightMin就可以直接判断了，复杂度是O(N)

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cctype>
using namespace std;

int N;
int a[100000]; 

void solve()
{
    int leftMax[N], rightMin[N];

    //leftMax[i]表示a中下标[0, i)的最大值
    leftMax[0] = 0;
    for(int i = 1; i < N; i ++){
        leftMax[i] = max(leftMax[i - 1], a[i - 1]);
    }

    //rightMin[i]表示a中下标(i, N-1]的最小值
    rightMin[N - 1] = (int)(1E9) + 1;
    for(int i = N - 2; i >= 0; i --){
        rightMin[i] = min(rightMin[i + 1], a[i + 1]);
    }

    int ans = 0, b[N];
    for(int i = 0; i < N; i ++){
        if(leftMax[i] <= a[i] && a[i] <= rightMin[i])
            b[ans ++] = a[i];
    }

    cout << ans << endl; 
    for(int i = 0; i < ans; i ++){
        cout << b[i];
        if(i + 1 < ans)
            cout << " ";
    }
    cout << endl; 
}

int main()
{
    cin >> N;
    for(int i = 0; i < N; i ++){
        cin >> a[i];
    }
    solve();
    return 0;
}

---

这里还有另外一种方法

我们先计算出lefMax[i]，并将排序好的数列放在B
A[i]只要满足一下2个条件就是主元：
1. leftMax[i] <= A[i]
2. B[i] == A[i]

证明：
条件1满足则左边的数都不大于A[i]，条件2满足说明排好序之后右边没有小于A[i]的数，即右边的数都不小于A[i]

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <cctype>
using namespace std;

int N;
int A[100000];

void solve()
{
    int B[N], leftMax[N];
    B[0] = A[0], leftMax[0] = 0;
    for(int i = 1; i < N; i ++){
        B[i] = A[i];
        leftMax[i] = max(leftMax[i - 1], A[i - 1]);
    }
    sort(B, B + N);

    int ans = 0, res[N];
    for(int i = 0; i < N; i ++){
        if(A[i] >= leftMax[i] && A[i] == B[i]){
            res[ans ++] = A[i];
        }
    }

    cout << ans << endl;
    for(int i = 0; i < ans; i ++){
        cout << res[i];
        if(i + 1 < ans)
            cout << " ";
    }
    cout << endl;
}

int main()
{
    cin >> N;
    for(int i = 0; i < N; i ++){
        cin >> A[i];
    }
    solve();
    return 0;
}