1098 Insertion or Heap Sort

题目大意：

给你初始序列和一段经过部分排序后的序列，问是经过了堆排序还是插入排序，输出排序方式并输出该排序下的下一步的输出序列。

解题思路：

模拟堆和插入排序的步骤即可，插入用sort更方便一些。每次完成一步后和目标序列进行比较，相同就再进行一步然后返回输出即可。
代码如下：

#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int n,origin[110],change[110],target[110];
bool same()//判断两个数组是否相同
{
  for(int i=1;i<=n;i++)
  {
    if(change[i]!=target[i])return false;
  }
  return true;
}
bool insertsort()//直接用sort排了
{
  for(int i=2;i<=n;i++)
  {
    sort(change,change+i+1);
    if(same())
    {
      //如果得到了目标序列，再排一次
      sort(change,change+i+2);
      return true;
    }
  }
  return false;
}
void downadjust(int i,int m)//最大堆向下调整
{
  int j=2*i;
  while(j<=m)//存在子节点
  {
    int tmp;//保存最大子节点的下标
    if((j+1)<=m)//如果有右节点
    {
      if(change[j]>change[j+1])tmp=j;
      else tmp=j+1;
    }
    else tmp=j;
    if(change[i]<change[tmp])
    {
      swap(change[i],change[tmp]);//交换堆顶与堆尾内容
      i=tmp;
      j=2*i;
    }
    else return;
  }
  return;
}
bool heapsort()
{
  for(int i=n/2;i>=1;i--)downadjust(i,n);//建堆（最大堆）
  for(int i=n;i>1;i--)//堆排序
  {
    swap(change[i],change[1]);
    downadjust(1,i-1);
    if(same())
    {//如果得到了目标序列，再排一次
      swap(change[i-1],change[1]);
      downadjust(1,i-2);
      return true;
    }
  }
  return false;
}
int main()
{
//  std::ios::sync_with_stdio(false);
//  cin.tie(0);
//  freopen("test.txt","r",stdin);
//  freopen("output.txt","w",stdout);
 scanf("%d",&n);
 for(int i=1;i<=n;i++)
 {
   scanf("%d",&origin[i]);
   change[i]=origin[i];
 }
 for(int i=1;i<=n;i++)
   scanf("%d",&target[i]);
 if(insertsort())//如果是插入排序
 {
   printf("Insertion Sort\n");
   printf("%d",change[1]);
   for(int i=2;i<=n;i++)printf(" %d",change[i]);
 }
 else//堆排序
 {
   for(int i=1;i<=n;i++)change[i]=origin[i];//还原备份数组
   heapsort();
   printf("Heap Sort\n");
   printf("%d",change[1]);
   for(int i=2;i<=n;i++)printf(" %d",change[i]);
 }
 return 0;
}