【PAT甲级、归并排序、折半插入排序、C++】1089 Insert or Merge (25分)

# include <iostream>
# include <vector>
# include <algorithm>
using namespace std;

vector<int> A;      // 折半插入排序所用的数组
vector<int> A_cpy;  // 归并排序所用的数组
vector<int> A_S;    // 题目所给的中间数组
int N;

// 输出折半插入排序所用的数组
void showA()
{
    for(int i=0;i<N;++i){
        cout << A[i];
        if(i != N-1)
            cout << " ";
    }cout << endl;
}

// 输出归并排序所用的数组
void showA_cpy()
{
    for(int i=0;i<N;++i){
        cout << A_cpy[i];
        if(i != N-1)
            cout << " ";
    }cout << endl;
}

// 折半插入排序
bool InsertSort()
{
    bool flag = false;
    for(size_t i=1;i<A.size();++i)
    {
        if(A == A_S && i !=1){
            cout << "Insertion Sort" << endl;
            flag = true;
        }

        // 二分插入排序主算法(可以用sort函数)
        // if(A[i] < A[i-1])
        // {
        //     int x = A[i];
        //     int pos = upper_bound(A.begin(), A.begin()+i-1, x) - A.begin();
        //     for(int j = i-1;j >= pos;--j)
        //         A[j+1] = A[j];
        //     A[pos] = x;
        // }
        sort(begin(A), begin(A)+i+1);
        
        if(flag == true){
            showA();
            return true;
        }
    }
    
    return false;
}


// 归并排序的递归形式控制输出有点难……
// bool MergeSort(int l, int r)
// {
//     bool flag = false;
//     if(l < r)
//     {
//         int mid = (l + r) / 2;
//         MergeSort(l, mid);
//         MergeSort(mid+1, r);
//         sort(A.begin()+l, A.begin()+r);
//     }
//     return false;
// }

// 区间合并
void Merge(int L1, int R1, int L2, int R2)
{
    int i = L1;
    int j = L2;
    vector<int> temp;
    while(i <= R1 && j <= R2)
    {
        if(A_cpy[i] <= A_cpy[j])
            temp.push_back(A_cpy[i++]);
        else
            temp.push_back(A_cpy[j++]);
    }
    while(i <= R1) temp.push_back(A_cpy[i++]);
    while(j <= R2) temp.push_back(A_cpy[j++]);
    
    for(size_t k=0;k<temp.size();k++)
        A_cpy[L1+k] = temp[k];
}
// 归并排序
bool MergeSort()
{
    bool flag = false;
    // 这里step是要和并的两个子区间的总长度，而不是一个自区间的长度
    for(int step = 2;step/2 <= N;step *= 2)
    {
        if(A_cpy == A_S && step != 2){
            cout << "Merge Sort" << endl;
            flag = true;
        }
        
        // 对某一趟的每个区间进行排序(用sort函数代替merge)
        for(int i = 0;i < N;i += step)
          sort(A_cpy.begin()+i, A_cpy.begin()+min(i+step, N));  // 不写merge函数用sort函数代替
		
		// 自己写一个merge函数
/*
        for(int i=0;i<N;i+=step)
        {
            int l1 = i, r1 = l1+step/2 - 1;
            int l2 = r1 + 1, r2 = l2+step/2 - 1;
            if(l2 < N) // 如果右子区间存在
                Merge(l1, r1, l2, min(r2, N-1));
        }
*/
            
        if(flag == true){
            showA_cpy();
            return true;
        }
    }
    return false;
}

# if 0
bool MergeSort()
{
    bool flag = false;
    // 这里的step是合并两个子区间中的一个子区间的长度
    for(int step = 1;step < N;step *= 2)
    {
        if(A_cpy == A_S && step != 1){
            cout << "Merge Sort" << endl;
            flag = true;
        }
        
        // 对某一趟的每个区间进行排序
//         for(int i = 0;i < N;i += 2*step)
//           sort(A_cpy.begin()+i, A_cpy.begin()+min(i+2*step, N));  // 不写merge函数用sort函数代替
		
		// 合并两个子区间
		for(int i=0;i<N;i += 2*step) {
            int l1 = i, r1 = l1+step - 1;
            int l2 = r1+1, r2 = min(l2+step - 1, N-1);
            if(l2 < N) // 如果右区间存在
                Merge(l1, r1, l2, r2);
        }
            
        if(flag == true){
            showA_cpy();
            return true;
        }
    }
    return false;
}
# endif




int main()
{
	// 输入
    cin >> N;
    for(int i=0;i<N;++i){
        int num;
        cin >> num;
        A.push_back(num);
        A_cpy.push_back(num);
    }
    for(int i=0;i<N;++i){
        int num;
        cin >> num;
        A_S.push_back(num);
    }
    
    // 排序输出
    if(MergeSort())
        return 0;
    else
        InsertSort();
    
    
    return 0;
}