PTA Replacement Selection (30 分)

Replacement Selection (30 分)

When the input is much too large to fit into memory, we have to do external sorting instead of internal sorting. One of the key steps in external sorting is to generate sets of sorted records (also called runs) with limited internal memory. The simplest method is to read as many records as possible into the memory, and sort them internally, then write the resulting run back to some tape. The size of each run is the same as the capacity of the internal memory.

Replacement Selection sorting algorithm was described in 1965 by Donald Knuth. Notice that as soon as the first record is written to an output tape, the memory it used becomes available for another record. Assume that we are sorting in ascending order, if the next record is not smaller than the record we have just output, then it can be included in the run.

For example, suppose that we have a set of input { 81, 94, 11, 96, 12, 99, 35 }, and our memory can sort 3 records only. By the simplest method we will obtain three runs: { 11, 81, 94 }, { 12, 96, 99 } and { 35 }. According to the replacement selection algorithm, we would read and sort the first 3 records { 81, 94, 11 } and output 11 as the smallest one. Then one space is available so 96 is read in and will join the first run since it is larger than 11. Now we have { 81, 94, 96 }. After 81 is out, 12 comes in but it must belong to the next run since it is smaller than 81. Hence we have { 94, 96, 12 } where 12 will stay since it belongs to the next run. When 94 is out and 99 is in, since 99 is larger than 94, it must belong to the first run. Eventually we will obtain two runs: the first one contains { 11, 81, 94, 96, 99 } and the second one contains { 12, 35 }.

Your job is to implement this replacement selection algorithm.

Input Specification:

Each input file contains several test cases. The first line gives two positive integers N (≤10^5) and M (<N/2), which are the total number of records to be sorted, and the capacity of the internal memory. Then N numbers are given in the next line, all in the range of int. All the numbers in a line are separated by a space.

Output Specification:

For each test case, print in each line a run (in ascending order) generated by the replacement selection algorithm. All the numbers in a line must be separated by exactly 1 space, and there must be no extra space at the beginning or the end of the line.

Sample Input:

13 3
81 94 11 96 12 99 17 35 28 58 41 75 15

Sample Output:

11 81 94 96 99
12 17 28 35 41 58 75
15

#include<bits/stdc++.h>
using namespace std;
priority_queue <int,vector<int>,greater<int> > q;
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    priority_queue <int,vector<int>,greater<int> > tmp;
    vector<int> ans;
    while(n--)
    {
        int x;
        scanf("%d",&x);
        if(q.size()<m) q.push(x);
        else{
            while(m)
            {
                if(ans.size()&&q.top()<ans.back())
                {
                    tmp.push(q.top());
                    q.pop();
                    m--;
                }
                else{
                    ans.push_back(q.top());
                    q.pop();
                    q.push(x);
                    break;
                }
            }
            if(m==0) 
            {
                q=tmp;
                while(!tmp.empty()) tmp.pop();
                m=q.size();
                for(int i=0;i<ans.size();++i)
                {
                    if(i==0) cout << ans[i];
                    else cout << " " << ans[i];
                }
                ans.clear();
                puts("");
                ans.push_back(q.top());
                q.pop();
                q.push(x);
            }
        }
    }
    while(!q.empty())
    {
        if(ans.size()&&q.top()<ans.back())
        {
            tmp.push(q.top());
            q.pop();
        }
        else{
            ans.push_back(q.top());
            q.pop();
        }
    }
    for(int i=0;i<ans.size();++i)
    {
        if(i==0) cout << ans[i];
        else cout << " " << ans[i];
    }
    puts("");
    int f=1;
    while(!tmp.empty())
    {
        if(f) cout << tmp.top();
        else cout << " " << tmp.top();
        tmp.pop();
        f=0;
    }
}

测试点5：存在相等的数，输出的序列不是严格升序的。