PAT A1098

题意

给出n和n个数的序列a和b，a为原始列列，b为排序其中的一个步骤，问b是a经过了堆排序还是插入排序的，并且输出它的下一步。

分析

① 插入排序的特点是：b 数组前面的顺序是从小到大的，后面的顺序不一定，但一定和原序列后面的顺序相同，所以只要遍历一下前面几位，遇到不是从小到大的时候，就判断 b 和 a 是不是对应位置的值相等，相等就说明是插入排序，否则就是堆排序。插入排序的下一步就是把第一个不符合从小到大的顺序的那个元素插入到前面已排序的序列的合适位置，那么只要对前几个已排序的后面一位 这个序列sort排序即可。
② 堆排序的特点是后面是从小到大的，前面的顺序不一定，那么就可以扫描到第一个不是前面所有序列最大值的那个数字 i，把它和第一个数字交换，然后把1~i-1进行一次向下调整，i 和 n 是需要调整的区间，因为是大顶堆，就是不断比较当前结点和自己的孩子结点哪个大，如果孩子大就把孩子结点和自己交换，然后再不断调整直到到达区间的最大值直到不能再继续了了为止。

AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[1005], b[1005];
void siftDown(int i, int n)
{
	int t, flag = 0;
	while (i * 2 <= n && flag == 0)
	{
		if (b[i] < b[i * 2])
		{
			t = i * 2;
		}
		else
		{
			t = i;
		}
		if (i * 2 + 1 <= n)
		{
			if (b[t] < b[i * 2 + 1])
			{
				t = i * 2 + 1;
			}
		}
		if (t != i)
		{
			swap(b[t], b[i]); 
			i = t;
		}
		else
		{
			flag = 1;
		}
	}
}

int main()
{
	int n, p = 2;
	scanf("%d", &n);
	for (int i = 1; i <= n; i++)
	{
		scanf("%d", &a[i]);
	}
	for (int i = 1; i <= n; i++)
	{
		scanf("%d", &b[i]);
	}
	while (p <= n && b[p - 1] <= b[p])
	{
		p++;
	}
	int index = p;
	while (p <= n && a[p] == b[p])
	{
		p++;
	}
	if (p == n + 1)
	{
		puts("Insertion Sort");
		sort(b + 1, b + 1 + index);
		for (int i = 1; i <= n; i++)
		{
			printf("%d", b[i]);
			if (i != n)
			{
				printf(" ");
			}
		}
	}
	else
	{
		puts("Heap Sort");
		p = n;
		while (p >= 2 && b[p] >= b[p - 1])
		{
			p--;
		}
		swap(b[1], b[p]);
		siftDown(1, p - 1);
		for (int i = 1; i <= n; i++)
		{
			printf("%d", b[i]);
			if (i != n)
			{
				printf(" ");
			}
		}
	}
	return 0;
}