本文共 2297 字,大约阅读时间需要 7 分钟。
给定两个有序整数数组 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--]; }}
转载地址:http://dsuvi.baihongyu.com/