合并两个有序数组
此博客链接:
合并两个有序数组
题目链接:https://leetcode-cn.com/problems/merge-sorted-array/
题目
给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。
初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。你可以假设 nums1 的空间大小等于 m + n,这样它就有足够的空间保存来自 nums2 的元素。
示例 1:
输入:nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3
输出:[1,2,2,3,5,6]
示例 2:
输入:nums1 = [1], m = 1, nums2 = [], n = 0
输出:[1]
题解
这里一开始想着在重新创建一个数组,把num1的非0元素先存起来,但是这样太耗费空间了。然后看了题解,有个思路挺好,从后向前插入元素,这样就不会修改num1前面的的数据。这里思路是,判断num1和num2中的元素那个大,就把那个元素放到num1中的最后,直到Num2的元素全部都被放入到num1中即完成。
代码
class Solution { public void merge(int[] nums1, int m, int[] nums2, int n) { if(m==1&&n==0) { return ; } if(m==0&&n!=0) { for(int i=0;i<n;i++) { nums1[i]=nums2[i]; } return ; } int len=nums2.length; int i=m-1; int j=n-1; int l=nums1.length-1; while(j>=0){ if(nums1[i]>nums2[j]) { nums1[l]=nums1[i]; if(i==0) { nums1[i]=0; } else i--; } else { nums1[l]=nums2[j]; j--; } l--; } } }
结果
[0,0,3,0,0,0,0,0,0] 3 [-1,1,1,1,2,3] 6
出来混总是要还的