PAT（甲级）2020年春季考试 7-4 Replacement Selection (30分)

前言

本人做法（16分,过了前两个测试点，第三个测试点错误，最后三个测试点超时）

//
// Created by niko on 2020/9/22.
//
//16分
#include <bits/stdc++.h>

using namespace std;
/**
 * 题意：首先给定n个待排序的数，以及capacity容量，
 * 首先按照每M个元素分组，分成n/m组（向上取整），随后从第一个组开始，选出最小的数minNow输出到第一个run中，
 * 随后按顺序从第二个组中选出该组中的第一个元素temp，若temp>=minNow，则temp最后一定是会输出到第一个run中的，
 * 若temp<minNow，那么保留temp来输出到下一个run（第二个run，因为本题要求每一个run中都是升序排序，而你刚刚输出到第一个
 * run中的元素比temp要大，那么显然temp肯定不能输出到第一个run中，只能输出到下一个run中），而保留temp的意思是后面继续输出到一个
 * run中的元素是除temp以外的其他元素中最小的元素，直到容量m中所有元素都属于下一个run，那么当前run输出结束，进入下一个循环。
 * @return
 */
const int N=1e5+10;
struct Node {
    int level;//层级（第level个run）
    int value;
} node[N];

bool cmp(Node a, Node b){
    if(a.level!=b.level)return a.level < b.level;
    return a.value < b.value;
}
vector<int> ans;
void printAns(){
    for (int i = 0, l = ans.size(); i < l; i++) {
        if(i!=0)printf(" ");
        printf("%d", ans[i]);
    }
    cout<<endl;
    ans.clear();
}
int main(){
    int n,m;
    cin>>n>>m;
    int nowLevel=1,index=m;//nowLevel表示当前输出到哪个run，index表示目前的下标
    vector<Node> v;
    for (int i = 0,data; i < n; ++i) {
        scanf("%d", &data);
        node[i] = {1, data};
        if (i < m) {
            v.push_back(node[i]);
        }
    }

    while (index < n||v.size()!=0) {
        Node temp;
        if (index<n){
            temp=node[index];//temp为待进入的元素
        }else{
            sort(v.begin(), v.end(), cmp);
            int i;
            for (i = 0; i < m; ++i) {
                if(v[i].level==nowLevel){
                    ans.push_back(v[i].value);
                } else{
                    printAns();
                    break;
                }
            }
            for (int j = i; j < m; ++j) {
                if(j!=i)printf(" ");
                printf("%d", v[j].value);
            }
            break;
        }
        sort(v.begin(), v.end(), cmp);//此时v[0]就是要输出的结点
        Node minNow = v[0];
        if (minNow.level == nowLevel + 1) {//上一个run已经输出完毕，现在开始进入下一个run
            printAns();
            nowLevel++;
            //更新后续所有结点level
            for (int i = index; i < n; ++i) {
                node[i].level = nowLevel;
            }
            temp.level = nowLevel;
        }
        if (minNow.value <= temp.value) {
            ans.push_back(minNow.value);
            v.erase(v.begin());
            v.push_back(temp);
        }else{
            temp.level = minNow.level + 1;
            ans.push_back(minNow.value);
            v.erase(v.begin());
            v.push_back(temp);
        }
        index++;
    }
    return 0;
}

AC代码（转自:https://blog.csdn.net/qq_39072627/article/details/107582244）

//致谢：https://blog.csdn.net/qq_39072627/article/details/107582244
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
int main() {
    int N, M;
    scanf("%d%d", &N, &M);
    vector<int> arr(N);
    for (int i = 0; i < N; i++) scanf("%d", &arr[i]);
    priority_queue<int, vector<int>, greater<int>> q;//模拟内存的排序，从小到大排列
    vector<int> v,line;//v存放下轮的数；line存放本轮的数
    int index = 0,count=0, last;
    for (; index < M; index++) q.push(arr[index]);
    while (count != N) {//如果从队列里出来的数小于N
        last = q.top();
        line.push_back(last);//根据有序性，每次列头的数会决定arr的下一个数去哪
        q.pop();
        count++;
        if (index < N) {//如果arr里还有数未处理
            if (arr[index] > last) q.push(arr[index++]);
            else v.push_back(arr[index++]);
        }
        if (q.empty()) {//当本轮的数已处理结束，即q为空
            for (int i = 0; i < line.size(); i++) {//输出line
                if (i != 0) printf(" ");
                printf("%d", line[i]);
            }
            printf("\n");
            line.clear();
            for (int i = 0; i < v.size(); i++)//v的数全都放到q里
                q.push(v[i]);
            v.clear();
        }
    }
}