题目:
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 n respectively.
Tag:
Array; Two Pointers
体会:
这个小题,初看很简单,但是实现起来还是有些小地方,需要特别注意。首先一个特别的地方就是在A里面merge B,然后从头开始merge,会出现需要插入操作,然后把A剩下的部分后移,因为不是链表,明显不合适。但是题目又要求merge到A, 那怎么办呢,解决办法就是从A的屁股上开始merge,就是从大数到小数开始merge。
具体执行的时候,我们的目标就是把Bmerge 完,即int j = n - 1, while (j >= 0)就继续执行。
那什么时候选择来自B的元素呢,这里面要注意A的边界情况,即A的元素已经都用完了,所以会有两种情况选择来自B的元素:
一是A的元素用完了 (i.e. i < 0),无条件选择B的;二是当A没完时,有B[j] > A[i]。
1 class Solution { 2 public: 3 void merge(int A[], int m, int B[], int n) { 4 int i = m - 1; 5 int j = n - 1; 6 int k = m + n - 1; 7 // when B is merged, job done 8 while (j >= 0) { 9 // merge 10 if (i < 0 || B[j] > A[i]) { 11 // when A is done or A is less than B, choose B 12 A[k] = B[j]; 13 j--; 14 } else { 15 A[k] = A[i]; 16 i--; 17 } 18 k--; 19 } 20 } 21 22 };
时间: 2024-10-11 01:24:35