3.3 面试题 10.01. 合并排序的数组

六眼飞鱼酱① 提交于 2020-03-03 13:50:13

题目

给定两个排序后的数组 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和B共有元素n个,那么目的就是有序的填慢A中的前n个空。
  • 用end表示A中要填满区间的尾index。endA表示A中元素的尾index, endB表示B的尾index。
  • 同时从尾部遍历A和B。end > -1为循环终止条件。
  • 当endA等于-1时,说明A中的元素遍历完了。若B中还有元素,填充到A中。
  • 当endB等于-1时,说明B中的已经填完了,此时直接break。
  • 当endA和endB都不为-1时,选择大的从尾部填充。

代码

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        int end = m + n - 1;
        int endA = m - 1;
        int endB = n - 1;

        while ( end > -1 ) {
            if ( endA == -1 ) 
                A[end--] = B[endB--];
            else if ( endB == -1 )
                break;
            else {
                if ( A[endA] >= B[endB] )
                    A[end--] = A[endA--];
                else 
                    A[end--] = B[endB--];
            }
        }

        return;
    }
};
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!