目录
题目描述:
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
- 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
- 你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
解题思路:
- 三个指针,分别指向两个数值的尾部,和合并后的数值尾部;
- 然后从尾部开始比较,把大的移动到nums1尾部,并且向前移动较大的指针;
- 直到其中一个移动完,再另一个数组剩下的部分追加到合并数组前面。
C++代码:
执行用时:0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗:8.8 MB, 在所有 C++ 提交中击败了64.24%的用户
通过测试用例:59 / 59
class Solution {
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
// 关键点,要求原地排序
// 用三个指针,分别指向两个数组的尾部,还有一个指向合并后数组尾部
// 从尾部开始比较,把大的移动到nums1尾部,并且向前移动较大的指针
// tail指向合并后数组尾部
int tail = m+n-1;
// m, n指向两个数组的尾部
m--;
n--;
while (m >= 0 && n >= 0)
{
if (nums1[m] > nums2[n])
{
nums1[tail] = nums1[m];
m--;
tail--;
}
else
{
nums1[tail] = nums2[n];
n--;
tail--;
}
}
while (m >= 0)
{
nums1[tail] = nums1[m];
m--;
tail--;
}
while (n >= 0)
{
nums1[tail] = nums2[n];
n--;
tail--;
}
}
};
python
执行用时:16 ms, 在所有 Python 提交中击败了76.34%的用户
内存消耗:13.1 MB, 在所有 Python 提交中击败了44.18%的用户
通过测试用例:59 / 59
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.
"""
# 定义三个指针
tail = m + n -1
m -= 1
n -= 1
while m >=0 and n >=0:
if nums1[m] > nums2[n]:
nums1[tail] = nums1[m]
m -= 1
tail -= 1
else:
nums1[tail] = nums2[n]
n -= 1
tail -= 1
while m >= 0:
nums1[tail] = nums1[m]
m -= 1
tail -= 1
while n >= 0:
nums1[tail] = nums2[n]
n -= 1
tail -= 1