蓝桥杯 经典算法题 合并排序数组

题目:

题解:

leetcode上也有这道题一模一样。和归并排序的小过程基本一模一样,只不过因为题目要求只能将arr2中元素合并到arr1中,一种可行的方法是按元素从大到小,顺序从每个序列尾部开始操作,第一填的位置是arr1的尾部,第一个判断的是arr1中有元素的尾部和arr2中有元素的尾部,和归并一模一样的过程构造出一个单调的数组。

#include <bits/stdc++.h>
using namespace std;
int main(){
  int a,b;
  cin>>a>>b;
  vector<int>arr1(a),arr2(a-b);
  for(int i=0;i<b;i++)cin>>arr1[i];
  for(int i=0;i<a-b;i++)cin>>arr2[i];
  int x=b-1,y=a-b-1,k=a-1;
  while(x>=0||y>=0){
    if(y<0||(x>=0&&arr1[x]>=arr2[y])){
      arr1[k--]=arr1[x--];
    }else {
      arr1[k--]=arr2[y--];
    }
  }
  for(auto x:arr1){
    cout<<x<<" ";
  }
  return 0;
}