题目描述
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
解法一,双指针
public void merge(int[] nums1, int m, int[] nums2, int n) {
int p1 = 0, p2 = 0;//两个指针分别从两个数组头开始
int[] sorted = new int[m + n];//辅助空间
int cur;
while (p1 < m || p2 < n) {
if (p1 == m) {//p1到头了
cur = nums2[p2++];//当前要添加的是p2下一个元素,并且p2更新
} else if (p2 == n) {//p2到头了
cur = nums1[p1++];//当前要添加的是p1下一个元素,并且p1更新
} else if (nums1[p1] < nums2[p2]) {//p1是小的
cur = nums1[p1++];//p1为待添加元素,并且p1更新
} else {//p2是小的
cur = nums2[p2++];//p2为待添加元素,并且p2更新
}
sorted[p1 + p2 - 1] = cur;//将cur加入
}
for (int i = 0; i != m + n; ++i) {//辅助数组转入原来nums1
nums1[i] = sorted[i];
}
这个思路我是有的,归并排序里用到,没写出来,笨比!!
用while不是for
解法二 直接合并排序
这个思路更简单。第二个加到第一个尾部。排序。抛弃了原本有序的性质。但代码简单
public:
void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
for (int i = 0; i != n; ++i) {
nums1[m + i] = nums2[i];
}
sort(nums1.begin(), nums1.end());
}