首页 > 代码库 > LeetCode: Merge Sorted Array [088]

LeetCode: Merge Sorted Array [088]

2024-07-06 17:38:14   224人阅读

【题目】


Given two sorted integer arrays A and B, merge B into A as one sorted array.

Note:
You may assume that A has enough space (size that is greater or equal to m + n) to hold additional elements from B. The number of elements initialized in A and B are m andn respectively.




【题意】

    给定两个有序的数组A和B，把两个数组合并成一个数组。
    注意，本题要求把B合并到A中去，并保证A的空间足够大。


【思路】

    如果按照以往处理链表合并或者申请新空间来合并的一贯做法，即从头开始依次合并，在本题的环境下会造成A中元素的大量移动。
    我们注意到A中有足够的空间，A数组后面的空间可以利用起来。我们换个方向，从数组尾部向头部合并，这样就避免了A中元素的移动。


【代码】

class Solution {
public:
    void merge(int A[], int m, int B[], int n) {
        int pa=m-1;
        int pb=n-1;
        int p=n+m-1;
        while(pa>=0 && pb>=0){
            if(A[pa]>B[pb]){
                A[p--]=A[pa--];
            }
            else{
                A[p--]=B[pb--];
            }
        }
        while(pb>=0){
            A[p--]=B[pb--];
        }
    }
};


联系
我们