题目
给你两个按非递减顺序排列的整数数组nums1和nums2,另有两个整数m和n,分别表示nums1和nums2中的元素数目。
请你合并nums2到nums1中,使合并后的数组同样按非递减顺序排列。
注意:最终,合并后数组不应由函数返回,而是存储在数组nums1中。为了应对这种情况,nums1的初始长度为m+n,其中前m个元素表示应合并的元素,后n个元素为0,应忽略。nums2的长度为n。
示例1:
输入:nums1=[1,2,3,0,0,0],m=3,nums2=[2,5,6],n=3
输出:[1,2,2,3,5,6]
解释:需要合并[1,2,3]和[2,5,6]。
合并结果是[1,2,2,3,5,6],其中斜体加粗标注的为nums1中的元素。
示例2:
输入:nums1=[1],m=1,nums2=[],n=0
输出:[1]
解释:需要合并[1]和[]。
合并结果是[1]。
示例3:
输入:nums1=[0],m=0,nums2=[1],n=1
输出:[1]
解释:需要合并的数组是[]和[1]。
合并结果是[1]。
注意,因为m=0,所以nums1中没有元素。nums1中仅存的0仅仅是为了确保合并结果可以顺利存放到nums1中。
代码
import java.util.Arrays;
public class Solution {
public static void merge(int[] nums1, int m, int[] nums2, int n) {
//利用传入的mn作指针
m--;
n--;
for (int i = nums1.length - 1; i >= 0; i--) {
if(m<0){
nums1[i] = nums2[n];
n--;
} else if (n<0) {
nums1[i] = nums1[m];
m--;
}else if (nums1[m] <= nums2[n]) {
nums1[i] = nums2[n];
n--;
} else {
nums1[i] = nums1[m];
m--;
}
}
}
public static void main(String[] args) {
{
int[] nums1 = new int[]{1, 2, 3, 0, 0, 0};
int m = 3;
int[] nums2 = new int[]{2, 5, 6};
int n = 3;
merge(nums1, m, nums2, n);
System.out.println(Arrays.toString(nums1));
}
{
int[] nums1 = new int[]{1};
int m = 1;
int[] nums2 = new int[]{};
int n = 0;
merge(nums1, m, nums2, n);
System.out.println(Arrays.toString(nums1));
}
{
int[] nums1 = new int[]{0};
int m = 0;
int[] nums2 = new int[]{1};
int n = 1;
merge(nums1, m, nums2, n);
System.out.println(Arrays.toString(nums1));
}
}
}