PAT甲级 1101 Quick Sort (25 分)

题目大意：

给定一个数组 \(a\) , 数组中有多少个元素 \(a_i\) 满足 \(a_i\) 大于等于左边的所有元素且 \(a_i\) 小于等于右边的所有元素。请输出满足元素的所有个数和元素。

思路：

求出两个数组 \(l[i]\) 和 \(r[i]\), 其中 $ l[i]$ 表示当前 \(i\) 左边的最大值， \(r[i]\) 表示当前 \(i\) 右边的最小值。然后依次判断就行了。

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 100010;

int n;
int a[N], l[N], r[N];
vector<int> v;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> n;
    int maxn = -1;
    for(int i = 0; i < n; i++){
       l[i] = maxn;
       cin >> a[i];
       maxn = max(maxn, a[i]);
    } 
    int minn = 0x3f3f3f3f;
    for(int i = n - 1; i >= 0; i--){
        r[i] = minn;
        minn = min(minn, a[i]);
    }
    int cnt = 0;
    for(int i = 0; i < n; i++){
        if(a[i] >= l[i] && a[i] <= r[i]) v.push_back(a[i]), cnt++;
        }
        
    cout << cnt << endl;
    if(cnt == 0) cout << endl;
    for(int i = 0; i < v.size(); i++)
        if(i != 0)cout << " " << v[i];
        else cout << v[i];
    return 0;
}