PAT乙级练习题B1035. 插入与归并

题目描述

根据维基百科的定义：

插入排序是迭代算法，逐一获得输入数据，逐步产生有序的输出序列。每步迭代中，算法从输入序列中取出一元素，将之插入有序序列中正确的位置。如此迭代直到全部元素有序。

归并排序进行如下迭代操作：首先将原始序列看成N个只包含1个元素的有序子序列，然后每次迭代归并两个相邻的有序子序列，直到最后只剩下1个有序的序列。

现给定原始序列和由某排序算法产生的中间序列，请你判断该算法究竟是哪种排序算法？

输入格式：

输入在第一行给出正整数N (<=100)；随后一行给出原始序列的N个整数；最后一行给出由某排序算法产生的中间序列。这里假设排序的目标序列是升序。数字间以空格分隔。

输出格式：

首先在第1行中输出“Insertion Sort”表示插入排序、或“Merge Sort”表示归并排序；然后在第2行中输出用该排序算法再迭代一轮的结果序列。题目保证每组测试的结果是唯一的。数字间以空格分隔，且行末不得有多余空格。
输入样例1：
10
3 1 2 8 7 5 9 4 6 0
1 2 3 7 8 5 9 4 6 0
输出样例1：
Insertion Sort
1 2 3 5 7 8 9 4 6 0
输入样例2：
10
3 1 2 8 7 5 9 4 0 6
1 3 2 8 5 7 4 9 0 6
输出样例2：
Merge Sort
1 2 3 8 4 5 7 9 0 6

题目解析

这里我使用sort()模拟插入排序和并归排序，其实就是排序的顺序不一样；注意排序循环判断的结束条件；

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

void print(vector<int> &ini)
{
    for (int i = 0; i < ini.size(); ++i)
    {
        cout << ini[i];
        if (i != ini.size() - 1)
        {
            cout << " ";
        }
        else
        {
            cout << endl;
        }
    }
    return;
}

bool is_insertion(vector<int> ini, vector<int> &mid)
{

    bool flag = 0;
    for (int i = 2; i <= ini.size();++i)
    {
        sort(ini.begin(), ini.begin()+i);
        if (flag)
        {
            cout << "Insertion Sort" << endl;
            print(ini);
            return 1;
        }
        if (ini == mid)
        {
            flag = 1;
        }
    }
    return 0;
}

void is_merge(vector<int> ini, vector<int> &mid)
{
    bool flag = 0;
    for (int i = 1; i/2 < ini.size(); i *= 2)
    {
        for (int j = 0, p = i; j <= ini.size() / i*i;j+=i,p=(ini.size()>p+i?p+i:ini.size()))
        {
            sort(ini.begin() + j, ini.begin() + p);
        }
        if (flag)
        {
            cout << "Merge Sort" << endl;
            print(ini);
            return;
        }
        if (ini == mid)
        {
            flag = 1;
        }
    }
    return;
}

int main()
{
    vector<int> ini,mid;
    int N;
    cin >> N;
    for (int i = 0; i < N; ++i)
    {
        int val;
        cin >> val;
        ini.push_back(val);
    }
    for (int i = 0; i < N; ++i)
    {
        int val;
        cin >> val;
        mid.push_back(val);
    }

    if (!is_insertion(ini, mid))
    {
        is_merge(ini, mid);
    }
    system("pause");
    return 0;
}