题目来源
88. 合并两个有序数组 - 力扣(LeetCode)
题目描述
给你两个按 非递减顺序 排列的整数数组 nums1
和 nums2
,另有两个整数 m
和 n
,分别表示 nums1
和 nums2
中的元素数目。
请你 合并 nums2
到 nums1
中,使合并后的数组同样按 非递减顺序 排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组 nums1
中。为了应对这种情况,nums1
的初始长度为 m + n
,其中前 m
个元素表示应合并的元素,后 n
个元素为 0
,应忽略。nums2
的长度为 n
。
示例
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
解释:需要合并 [1,2,3] 和 [2,5,6] 。合并结果是 [1,2,2,3,5,6] ,其中斜体加粗标注的为 nums1 中的元素。
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
解释:需要合并 [1] 和 [] 。合并结果是 [1] 。
示例 3:
输入:nums1 = [0], m = 0, nums2 = [1], n = 1
输出:[1]
解释:需要合并的数组是 [] 和 [1] 。合并结果是 [1] 。注意,因为 m = 0 ,所以 nums1 中没有元素。nums1 中仅存的 0 仅仅是为了确保合并结果可以顺利存放到 nums1 中。
提示
- nums1.length == m + n
- nums2.length == n
- 0 <= m, n <= 200
- 1 <= m + n <= 200
- -10^9 <= nums1[i], nums2[j] <= 10^9
题目解析
我们可以创建一个(长度m+n)临时数组tmp。然后定义三个指针 i, j ,k,其中 i 指向nums1的索引0位置,j 指向 nums2的索引0位置,k指向 tmp的索引0位置。
之后三个指针的移动逻辑如下:
比较nums1[i] 和 nums2[j] 的大小,较小的填入tmp[k]位置
由于本题要求最终将nums2并入nums1,因此我们需要将tmp数组元素覆盖进nums1数组元素。
Java算法源码
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int[] tmp = new int[m + n];
int i = 0;
int j = 0;
for (int k = 0; k < m + n; k++) {
if (i == m) {
tmp[k] = nums2[j++];
} else if (j == n) {
tmp[k] = nums1[i++];
} else if (nums1[i] <= nums2[j]) {
tmp[k] = nums1[i++];
} else {
tmp[k] = nums2[j++];
}
}
System.arraycopy(tmp, 0, nums1, 0, m + n);
}
}
JavaScript算法源码
/**
* @param {number[]} nums1
* @param {number} m
* @param {number[]} nums2
* @param {number} n
* @return {void} Do not return anything, modify nums1 in-place instead.
*/
var merge = function (nums1, m, nums2, n) {
let i = 0;
let j = 0;
const tmp = new Array(m + n);
for (let k = 0; k < m + n; k++) {
if (i == m) {
tmp[k] = nums2[j++];
} else if (j == n) {
tmp[k] = nums1[i++];
} else if (nums1[i] <= nums2[j]) {
tmp[k] = nums1[i++];
} else {
tmp[k] = nums2[j++];
}
}
for (let k = 0; k < m + n; k++) {
nums1[k] = tmp[k];
}
};
Python算法源码
class Solution(object):
def merge(self, nums1, m, nums2, n):
"""
:type nums1: List[int]
:type m: int
:type nums2: List[int]
:type n: int
:rtype: None Do not return anything, modify nums1 in-place instead.
"""
i = 0
j = 0
tmp = [0] * (m + n)
for k in range(m + n):
if i == m:
tmp[k] = nums2[j]
j += 1
elif j == n:
tmp[k] = nums1[i]
i += 1
elif nums1[i] <= nums2[j]:
tmp[k] = nums1[i]
i += 1
else:
tmp[k] = nums2[j]
j += 1
for k in range(m + n):
nums1[k] = tmp[k]
C算法源码
void merge(int* nums1, int nums1Size, int m, int* nums2, int nums2Size, int n) {
int i = 0;
int j = 0;
int tmp[m + n];
for (int k = 0; k < m + n; k++) {
if (i == m) {
tmp[k] = nums2[j++];
} else if (j == n) {
tmp[k] = nums1[i++];
} else if (nums1[i] <= nums2[j]) {
tmp[k] = nums1[i++];
} else {
tmp[k] = nums2[j++];
}
}
for (int k = 0; k < m + n; k++) {
nums1[k] = tmp[k];
}
}
C++算法源码
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
int i = 0;
int j = 0;
vector<int> tmp(m + n);
for (int k = 0; k < m + n; k++) {
if (i == m) {
tmp[k] = nums2[j++];
} else if (j == n) {
tmp[k] = nums1[i++];
} else if (nums1[i] <= nums2[j]) {
tmp[k] = nums1[i++];
} else {
tmp[k] = nums2[j++];
}
}
for (int k = 0; k < m + n; k++) {
nums1[k] = tmp[k];
}
}
};