PAT乙级别.1035. 插入与归并(25)

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

PAT链接

思路

搞清楚插入排序与归并排序的实现

代码

/**
* @tag     PAT_B_1035
* @authors R11happy (xushuai100@126.com)
* @date    2016-9-7 14:57-16:00
* @version 1.0
* @Language C++
* @Ranking  740/614
* @function null
*/

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
int origin[110],changed[110],temp[110],output[110];

//判断两个序列是否一样
bool equals(int A[], int B[], int len)
{
    for(int i = 0; i<len; i++)
    {
        if(A[i] != B[i])    return false;
    }
    return true;
}

bool insertSort(int A[], int len)
{
    bool flag = false;
    for(int i = 0; i<len; i++)
    {
        temp[i] = A[i];
    }
    //每趟排完，前i个数是有序的
    for(int i = 1; i<len; i++)
    {
        for(int j = i; j>0; j--)    //进行一趟插入排序
        {
            if(temp[j] < temp[j-1])   swap(temp[j], temp[j-1]);
        }
        if(flag == true)
        {
            for(int k = 0; k<len; k++)  output[k] = temp[k];
            return true;
        }
        if(equals(changed, temp, len))   flag = true;
    }
    return false;
}

void merge(int A[], int L1, int R1, int L2, int R2)
{
    int i = L1, j = L2;
    int store[110], index = 0;
    while(i <= R1 && j <= R2)
    {
        if(A[i] < A[j]) store[index++] = A[i++];
        else    store[index++] = A[j++];
    }
    while(i <= R1)  store[index++] = A[i++];
    while(j <= R2)  store[index++] = A[j++];
    for(int k = 0; k < index; k++)
    {
        A[L1+k] = store[k]; //千万注意，这里是A[L1+k]不是A[k]
    }
}

bool mergeSort(int A[], int len)
{
    bool flag = false;
    for(int i = 0; i<len; i++)
    {
        temp[i] = A[i];
    }

    for(int step = 2; step/2 < len; step *= 2)
    {
        for(int i = 0; i<len; i+=step)
        {
            int mid = i + step/2 -1;    //该组左半边最后一个元素
            if(mid + 1 <len)
            {
                merge(temp, i, mid, mid+1, min(len-1, i+step-1));
            }
        }
        if(flag == true)
        {
            for(int k = 0; k<len; k++)  output[k] = temp[k];
            return true;
        }
        if(equals(changed, temp, len))   flag = true;
    }
    return false;
}

int main(int argc, char const *argv[])
{
    int N;
    scanf("%d", &N);
    for(int i = 0; i<N; i++)
    {
        scanf("%d", &origin[i]);
    }
    for(int i = 0; i<N; i++)
    {
        scanf("%d", &changed[i]);
    }
    if(insertSort(origin, N))
    {
        printf("Insertion Sort\n");
        printf("%d", output[0] );
        for(int i = 1; i<N; i++)    printf(" %d",output[i] );
    }
    else if(mergeSort(origin, N))
    {
        printf("Merge Sort\n");
        printf("%d", output[0] );
        for(int i = 1; i<N; i++)    printf(" %d",output[i] );
    }
    else
        printf("Error\n");
    return 0;
}

收获