1098. Insertion or Heap Sort (25)--堆排序

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>

#include <string>
#include <cstring>
#include <vector>
#include <queue>

using namespace std;

int N;
int a[505];
int b[505];
int co[505];

void init()
{
    for(int i=0;i<505;i++)
    {
        a[i]=2100000000;
        b[i]=2100000000;
        co[i]=2100000000;
    }

}

void printa()
{
    for(int i=0;i<N;i++)
    {
       printf("%d",a[i]);
       if(i!=N-1)
            printf(" ");
    }
    printf("\n");
}

void printco()
{
    for(int i=0;i<N;i++)
    {
       printf("%d",co[i]);
       if(i!=N-1)
            printf(" ");
    }
    printf("\n");
}

bool judge(int a1[],int a2[])
{
    int i;
    for(i=0;i<N;i++)
    {
        if(a1[i]!=a2[i])
            break;
    }

    if(i==N)
        return true;
    else
        return false;
}


bool judge2(int a1[],int a2[])
{
    int i;
    for(i=1;i<=N;i++)
    {
        if(a1[i]!=a2[i-1])
            break;
    }

    if(i==N+1)
        return true;
    else
        return false;
}

bool insort()
{
    for(int i=1;i<N;i++)
    {
        int t=a[i];
        int p=i-1;
        while(p>=0 && a[p]>=t)
        {
            a[p+1]=a[p];
            p--;
        }
        a[p+1]=t;

        //printa();

        if(judge(a,b))
        {
            printf("Insertion Sort\n");

            i=i+1;
            int t=a[i];
            int p=i-1;
            while(p>=0 && a[p]>=t)
            {
                a[p+1]=a[p];
                p--;
            }
            a[p+1]=t;

            printa();
            return true;
        }
    }

    return false;
}

void downadj(int low,int high)
{
    int i=low;
    int j=i*2;

    while(j<=high)
    {
        if(j+1<=high&&a[j+1]>a[j])
            j=j+1;

        if(a[j]>a[i])
        {
            swap(a[i],a[j]);
            i=j;
            j=i*2;
        }
        else
            break;
    }

}

void creat()
{
    for(int i=N/2;i>=1;i--)
    {
        downadj(i,N);
    }
}

void heapsort()
{
    for(int i=1;i<=N;i++)//复原a数组
        a[i]=co[i-1];
    creat();

    int i;
    for(i=N;i>1;i--)//i为堆的体积
    {
        swap(a[i],a[1]);
        downadj(1,i-1);
        if(judge2(a,b))
        {
            break;
        }
    }
    i--;
    swap(a[i],a[1]);
    downadj(1,i-1);
    for(int i=1;i<=N;i++)
    {
        printf("%d",a[i]);
        if(i!=N)
            printf(" ");
    }

}


int main()
{
//    freopen("in.txt","r",stdin);

    init();
    scanf("%d",&N);

    for(int i=0;i<N;i++)
    {
        int t;
        scanf("%d",&t);
        a[i]=t;
        co[i]=t;
    }

    for(int i=0;i<N;i++)
        scanf("%d",&b[i]);

    if(insort()==false)
    {
        printf("Heap Sort\n");
        heapsort();
    }



    return 0;
}