给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
1.nums1 = [1,2,3,0,0,0], m = 3
2.nums2 = [2,5,6], n = 3输出: [1,2,2,3,5,6]
直接比较nums1和
nums2元素的大小,然后根据大小加入到nums1的末尾,最后还要考虑nums2的元素是否还有剩余。设定三个变量i,j,k,分别指向nums1,nums2和混合数组的末尾。进行while循环,如果i和j都大于0,再看如果nums1[i] > nums2[j],说明要把nums1[i]先加入混合数组的末尾,加入后k和i都要自减1;else,就把nums2[j]加入混合数组的末尾,之后j和k也要自减1。循环结束后,有可能i和j还大于等于0,若j大于等于0,那么需要继续循环,将nums2中的数字继续拷入nums1。若是i大于等于0,那么就不用管,因为混合数组本身就存储在nums1中。
class Solution:
def merge(self, nums1, m, nums2, n):
"""
:type nums1: List[int]
:type m: int
:type nums2: List[int]
:type n: int
:rtype: void Do not return anything, modify nums1 in-place instead.
"""
i = m-1 #-1代表数组的末尾下标,因为下标是从0开始的
j = n-1
k = m+n-1
while(i>=0 and j>=0):
if(nums1[i] > nums2[j]):
nums1[k] = nums1[i]
k = k-1
i = i-1
else:
nums1[k] = nums2[j]
k = k-1
j = j-1
while(j>=0): # 如果nums2的元素还有剩余
nums1[k] = nums2[j]
k = k-1
j = j-1
注意:比较应该从后往前开始(从数组最后开始),不然如果从小往大排列那么数组1中的元素将有可能会被覆盖掉。
python语法中并不存在 i++ 或 i-- 这种自增或自减的语法
以下是Java版本:
题目:
给定两个排序整数数组A和B,将b合并为一个排序数组。
注意:您可以假设a有足够的空间(大小大于或等于m + n)来保存来自B的附加元素。
分析:
循环递归解决即可,最大的在最后,倒序进行处理
方法1:
public class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
// 特别注意这里的m只是nums1中元素的个数,不是最终第一个数组的长度 不要用m=nums1.length;不然会造成数据越界的报错
int i=m-1,j=n-1,index=m+n-1;
while(i>=0&&j>=0){
if(nums1[i]>nums2[j]){
//A大就把A的数组放在更后面
nums1[index--]=nums1[i--];
}
else{
nums1[index--]=nums2[j--];
}
}
while(i>=0){
//A大就把A的数组放在更后面
nums1[index--]=nums1[i--];
}
while(j>=0){
nums1[index--]=nums2[j--];
}
}
}
方法二:三行代码
public class Solution {
public void merge(int[] A, int m, int[] B, int n) {
int i=m-1, j=n-1, k=m+n-1;
while (i>-1 && j>-1) A[k--]= (A[i]>B[j]) ? A[i--] : B[j--];
while (j>-1)
A[k--]=B[j--];
}
}