数据结构 _ 基础练习 _ 7-14 Insertion or Heap Sort

原题

点此链接1

题目分析

  与前一题“Insert or Merge”相同2，同样考察的是插入排序算法以及堆排序算法。

  算法如下：

1. 首先需要判断是插入排序还是堆排序，由于插入必然是 “有序序列+相同的序列”，利用插入排序判别更佳
2. 若为插入排序，则只需要将下一个元素插入进已排序列即可
3. 若为堆排序，则需要判别目前的堆的大小，由于末尾是有序序列，可以根据末尾元素和首元素的大小关系来判别。随后根据课本（高等教育出版社 - 陈越主编 - 《数据结构》）P148最大堆创建算法来调整堆，此处直接调用标准库"make_heap"算法实现。

代码

/**
 * @file Insertion_or_HeapSort.cpp
 * @author your name (you@domain.com)
 * @brief https://pintia.cn/problem-sets/16/problems/676
 * @version 0.1
 * @date 2021-02-22
 * 
 * @copyright Copyright (c) 2021
 * 
 */

/*
 *  解题思路
 *  1.判断是插入排序还是堆排序
 *      插入排序必然是：有序序列 + 与原序列完全相同的序列
 *  2.若为插入排序，则对后一个元素进行插入算法
 *  3.若为堆排序
 *      计算目前的堆首和堆尾
 *      交换堆首和堆尾元素，并将堆尾-1
 *      生成新的最大堆
 */

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int main()
{
    int n;
    cin >> n;
    vector<int> d1;
    vector<int> d2;
    for (auto i = 0; i < n; ++i)
    {
        auto tmp = 0;
        cin >> tmp;
        d1.push_back(tmp);
    }
    auto flag = false;
    auto len = 0;
    auto flagSort = 0; // 0->insertion sort  1->heap sort
    for (auto i = 0; i < n; ++i)
    {
        auto tmp = 0;
        cin >> tmp;
        if (!flagSort)
        {
            if (!flag)
            {
                if (!d2.empty() && d2.back() > tmp)
                {
                    flag = true;
                    len = i;
                }
            }
            else
            {
                if (d1[i] != tmp)
                    flagSort = 1;
            }
        }
        d2.push_back(tmp);
    }

    if (!flagSort)
    {
        cout << "Insertion Sort" << endl;
        auto iter = d2.begin() + len;
        auto tmp = *iter;
        for (; iter != d2.begin() && *(iter - 1) > tmp; --iter)
            *iter = *(iter - 1);
        *iter = tmp;
    }
    else
    {
        cout << "Heap Sort" << endl;
        auto first = d2.front();
        auto iter = d2.end() - 1;
        for (; iter != d2.begin() && *iter >= first; --iter)
            ;
        *(d2.begin()) = *iter;
        *iter = first;
        make_heap(d2.begin(), iter);
    }

    auto iter = d2.begin();
    for (; iter != d2.end() - 1; ++iter)
        cout << *iter << " ";
    cout << *iter << endl;

    return 0;
}