原题
点此链接1
题目分析
与前一题“Insert or Merge”相同2,同样考察的是插入排序算法以及堆排序算法。
算法如下:
- 首先需要判断是插入排序还是堆排序,由于插入必然是 “有序序列+相同的序列”,利用插入排序判别更佳
- 若为插入排序,则只需要将下一个元素插入进已排序列即可
- 若为堆排序,则需要判别目前的堆的大小,由于末尾是有序序列,可以根据末尾元素和首元素的大小关系来判别。随后根据课本(高等教育出版社 - 陈越主编 - 《数据结构》)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;
}