PAT (Advanced Level) 1098. Insertion or Heap Sort (25)

简单题。判断一下是插排还是堆排。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<algorithm>
using namespace std;

const int maxn=;
int a[maxn],b[maxn],n;

int main()
{
    memset(b,-,sizeof b);
    scanf("%d",&n);
    for(int i=; i<=n; i++) scanf("%d",&a[i]);
    for(int i=; i<=n; i++) scanf("%d",&b[i]);

    int flag=;

    int p=n+;
    for(int i=; i<=n; i++)
        if(b[i]<b[i-])
        {
            p=i;
            break;
        }

    for(int i=p; i<=n; i++)
    {
        if(a[i]==b[i]) continue;
        else flag=;
    }

    if(flag==)
    {
        printf("Insertion Sort\n");
        if(p==n+) p=n;
        sort(a+,a+p+);
        for(int i=; i<=p; i++)
        {
            printf("%d",a[i]);
            if(i<n) printf(" ");
            else printf("\n");
        }
        for(int i=p+; i<=n; i++)
        {
            printf("%d",a[i]);
            if(i<n) printf(" ");
            else printf("\n");
        }
    }
    else
    {
        printf("Heap Sort\n");
        sort(a+,a++n);
        for(int i=n; i>=; i--)
        {
            if(a[i]==b[i]) p=i;
            else break;
        }

        swap(b[],b[p-]);

        int i=;;
        while()
        {
            if(*i<p-&&*i+<p-)
            {
                if(b[i]<max(b[*i],b[*i+]))
                {

                    if(b[*i]>b[*i+])
                    {
                        swap(b[*i],b[i]);
                        i=*i;
                    }
                    else
                    {
                        swap(b[*i+],b[i]);
                        i=*i+;
                    }
                }
                else break;
            }
            else if(*i<p-)
            {
                if(b[i]<b[*i])
                {
                    swap(b[*i],b[i]);
                    i=*i;
                }
            }
            else break;

        }
        for(int i=; i<=n; i++)
        {
            printf("%d",b[i]);
            if(i<n) printf(" ");
            else printf("\n");
        }
    }

    return ;
}