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

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

初始化 A 和 B 的元素数量分别为 mn

示例:

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

代码如下:

class Solution {
public:
    void merge(vector& A, int m, vector& B, int n) {
        int i=m-1,j=n-1,k=m+n-1;//定义下标
        while(i>=0&&j>=0)
        {
            if(A[i]>B[j])//比较两个数组里的哪个数字大,选择较大的数字放在数组里
            {
                A[k--]=A[i--];
            }
            else
            {
                A[k--]=B[j--];
            }
        }
        
        // 1.如果i不为0,则A没遍历完,都已经在A中不用再管
        // 2.如果j不为0,则B没遍历完,把B里面的数字移到A里
        /*while(i>=0)
        {
            A[k--]=A[i--];
        }*/
        while(j>=0)
        {
            A[k--]=B[j--];
        }

    }
};

你可能感兴趣的:(算法,leetcode,数据结构)