刷题-力扣-面试题 10.01. 合并排序的数组

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sorted-merge-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

给定两个排序后的数组 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. 根据题目描述,合并排序
  2. 从两个数组的末尾开始比较,较大的放置于A的末尾

代码

class Solution {
public:
    void merge(vector<int>& A, int m, vector<int>& B, int n) {
        int index = A.size() - 1;
        m -= 1;
        n -= 1;
        while (n >= 0 && m >= 0) {
            if (A[m] > B[n]) { A[index--] = A[m--]; }
            else { A[index--] = B[n--]; }
        }
        while (n >= 0) {
            A[index--] = B[n--];
        }
        return;
    }
};
posted @ 2022-03-07 12:43  韩亚光  阅读(27)  评论(0编辑  收藏  举报