PTA 乙级 1035 插入与归并 (25分) C++

插入排序：https://blog.csdn.net/qq_42453117/article/details/99680831

归并排序：https://blog.csdn.net/qq_42453117/article/details/100036347

测试点0、2、4：插入排序

测试点1、3、5、6：归并排序

C++

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

// 插入排序
int insertionsort(vector<int> a, vector<int> b, int n) {
    int flag = 0;
    for (int i = 1; i < n; i++) {            // 从arr[0]后的元素开始
        int value = a[i];                    // 定义待插入的数
        int index = i;                        // 找到待插入数的下标
        while (index > 0 && value < a[index - 1]) {    // 元素右移
            a[index] = a[index - 1];
            index--;
        }
        a[index] = value;        // 找到传入元素所在位置赋值
        if (flag) {                // 输出该排序算法再迭代一轮的序列
            cout << "Insertion Sort" << endl;
            cout << a[0];
            for (int j = 1; j < n; j++)cout << " " << a[j];
            return 1;
        }
        if (equal(a.begin(), a.end(), b.begin()))    // 与第二行数据进行比较
            flag = 1;
    }
    return 0;
}

// 归并排序
void mergesort(vector<int> a, vector<int> b,int n) {   
    int k = 1;                    // 分割大小
    int flag = 0;
    while (k <= n ) {            // 当分割的大小小于数组大小时
        for (int i = 0; i < n; i += k) {
            sort(a.begin() + i, a.begin() + (i + k < n ? i + k : n));    // 将分割的数组进行排序
        }
        k *= 2;                    //分割大小以二倍增大
        if (flag) {
            cout << "Merge Sort" << endl;
            cout << a[0];
            for (int j = 1; j < n; j++)cout << " " << a[j];
            return;
        }
        if (equal(a.begin(), a.end(), b.begin()))
            flag = 1;
    }
}

int main() {
    int n = 0;
    cin >> n;
    vector<int> arr(n);
    vector<int> asort(n);
    for (int i = 0; i < n; i++) cin >> arr[i];
    for (int i = 0; i < n; i++) cin >> asort[i];
    if (insertionsort(arr, asort, n)) exit(0);        //是插入排序直接退出
    mergesort(arr, asort, n);
    return 0;
}