[LeetCode] Merge Sorted Array

来源:互联网 发布:视频格式转换软件下载 编辑:程序博客网 时间:2024/05/23 12:21

题目:

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 and nrespectively.

代码:

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

思路:

如果企图从头到尾把B中的元素插入到A中,插入位置后面的所有元素整体后移的办法,那效率确实有些低。

本文所示代码的算法是从后往前插入,开始插入的位置为m+n-1,由此保证插入过程不会覆盖掉A原有的元素值。

0 0
原创粉丝点击