PAT乙集找到了一个比较有意思的题目:
1035. 插入与归并(25)
时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue
根据维基百科的定义:
插入排序是迭代算法,逐一获得输入数据,逐步产生有序的输出序列。每步迭代中,算法从输入序列中取出一元素,将之插入有序序列中正确的位置。如此迭代直到全部元素有序。
归并排序进行如下迭代操作:首先将原始序列看成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
最初的思路是直接模拟,然后每一次迭代的时候比较并记录结果。
写了一遍,AC失败……
最开始的问题是归并排序,我用的是自顶向下递归排序,后来认真读题才发现应该用自底向上的方法。
然后是插入排序部分,应该是没有考虑两数相等的情况,这个纯属手滑……
总之各种问题,后来参考了一下百度,发现插入排序这部分根本不用去模拟。
算法很巧妙,主要参考了这里的算法。
代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
int a[101] = {0};
int b[101] = {0};
int n=0;
int min(int &a, int &b){
if(a<b) return a;
else return b;
}
bool equal(){
for(int i=0;i<n;++i)
if(a[i]!=b[i])
return false;
return true;
}
void merge(){
bool is_merge = false;
for(int sz=1; ; sz+=sz){
is_merge=equal();
int lo=0;
for(; lo+sz<n; lo+=sz){
sort(a+lo, a+lo+sz);
}
sort(a+lo, a+n);
if(is_merge)
return;
}
}
int main(){
cin>>n;
for(int i=0;i<n;++i)
cin>>a[i];
for(int i=0;i<n;++i)
cin>>b[i];
bool is_insert = false;
int i,j;
for(i=0; b[i]<=b[i+1]&&i<n; ++i);//i及其之前下标处的数字都有序
for(j=++i; a[j]==b[j]&&j<n;++j);//i之后的数字都相同
if(j==n){
cout<<"Insertion Sort"<<endl;
sort(a, a+i+1);
is_insert = true;
}
else{
merge();
cout<<"Merge Sort"<<endl;
}
for(int i=0;i<n-1;++i)
cout<<a[i]<<" ";
cout<<a[n-1];
return 0;
}