1035. 插入与归并(25)

测试点最后两个有错误，希望知道的同仁能够指点一下，感谢！

根据维基百科的定义：

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

归并排序进行如下迭代操作：首先将原始序列看成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

#include "iostream"
#include "vector"
#include "algorithm"

using namespace std;
void insertSort(vector<int> in, int index)
{
	sort(in.begin(), in.begin() + index + 1);
	for (int i = 0; i < in.size(); i++)
	{
		if (i < in.size() - 1)
			cout << in[i] << " ";
		else
			cout << in[i];
	}
}

void mergeSort(vector<int> in, vector<int> target, int total)
{                    
	bool flag = true;
	
	int index = 1;
	int i;
	while (flag)
	{
		flag = false;
		if (!equal(in.begin(), in.end(), target.begin()))             // 判断是否和目标数组相同
			flag = true; 
		
		index *= 2;
		
		for (i = 0; i < total / index; i++)
		{
			sort(in.begin() + i*index, in.begin() + (i + 1)*index);  // 对前面的total/index对进行排序
		}
		
		sort(in.begin()+index*total/index, in.end());                        // 对剩下的进行排序
	}
	for (i = 0; i < in.size(); i++)
	{
		if (i < in.size() - 1)
			cout << in[i] << " ";
		else
			cout << in[i];
	}
}

int main()
{
	vector<int> input;
	vector<int> order;
	int tmp = 0;
	int N = 0;
	int index = 0;
	int index_insert = 0;
	int index_merge = 0;
	
	cin >> N;

	for (int i = 0; i < N; i++)
	{
		cin >> tmp;
		input.push_back(tmp);
	}

	for (int i = 0; i < N; i++)
	{
		cin >> tmp;
		order.push_back(tmp);
	}

	for (int i = 0; i < N-1; i++)
	{
		if (order[i] > order[i + 1])
		{
			index = i + 1;
			break;	
		}
	}
	index_insert = index;
	index_merge = index;
	for (int i = index; i < N; i++)      // 找到最后一个相同的数字的下标
	{
		 if (order[i] == input[i])
		 {
			 index = i;
		 }
		 else
		 {
			 index = i;      
			 break;
		 }
	}

	if (index == N - 1)
	{
		cout << "Insertion Sort" << endl;
		insertSort(order, index_insert);
	}
	else
	{
		cout << "Merge Sort" << endl;
		mergeSort(input, order, N);
	}

	system("pause");
	return 0;
}