【程序员面试金典】面试题 10.01. 合并排序的数组

题目描述

描述：给定两个排序后的数组 A 和 B，其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法，将 B 合并入 A 并排序。

初始化 A 和 B 的元素数量分别为 m 和 n。

示例:

输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6],       n = 3

输出: [1,2,2,3,5,6]

说明:

A.length == n + m

解题思路

思路1：最直观的想法是，三指针。虽然A和B都是有序数组，但是A的缓冲空间在尾部，如果从头开始遍历，则会导致过多的移动，故使用三指针从尾部开始遍历。

void merge(vector<int>& A, int m, vector<int>& B, int n) 
{
   int cur=m+n-1;    //指向合并后的数组尾部
   int i=m-1;    //指向A尾部
   int j=n-1;    //指向B尾部
   while(i>=0&&j>=0)  //同时遍历
   {
       if(A[i]>=B[j])
       {
          A[cur]=A[i];
          cur--;
          i--;
       }
       else
       {
          A[cur]=B[j];
          cur--;
          j--;
       }
   }
   while(i>=0)
   {
       A[cur]=A[i];
       cur--;
       i--;
   }
   while(j>=0)
   {
       A[cur]=B[j];
       cur--;
       j--;
   }
}

总结：注意遍历顺序，同时注意两个同时遍历完，还要考虑遍历剩余的那个。