题解 | 合并两个有序的数组

合并两个有序的数组

https://www.nowcoder.com/practice/89865d4375634fc484f3a24b7fe65665

从前往后扫描,直接在A内部操作,遇到一个比A[i]小的B[j], 就把A[i:]全部后移一格。

坑:

  1. 需要处理极端情况m==0和n==0。
  2. 需要重新初始化A[m:]为INT最大值,避免比较出问题。
#include <limits>
class Solution {
  public:
    void merge(int A[], int m, int B[], int n) {
        int i = 0, j = 0;
        if (m == 0) {
            for (int i = 0; i < n; i++) A[i] = B[i];
            return;
        }
        if (n == 0) return;
        for(int i = m; i < m + n; i++) A[i] = numeric_limits<int>::max();
        while ( i < m + n && j < n) {
            if ( A[i] <= B[j]) {
                i++;
            } else {
                // Move A[i:] by 1
                for (int k = m + j ; k > i ; k--) {
                    A[k] = A[k - 1];
                }
                A[i] = B[j];
                j++;
            }
        }
        while (j < n) A[i++] = B[j++];
    }
};

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务