此博客链接:
合并两个有序数组
题目链接:https://leetcode-cn.com/problems/merge-sorted-array/submissions/
题目
给你两个有序整数数组 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]
题解
先把nums1中的数组存到一个新的数组arr中,分别遍历arr和nums2中的元素取小的数组存入到nums1中。
代码
class Solution { public void merge(int[] nums1, int m, int[] nums2, int n) { int arr[]=new int[m]; for(int i=0;i<m;i++) { arr[i]=nums1[i]; } int p=0,q=0; int k=0; while(p<m||q<n) { while(p<m&&q<n){ nums1[k++]=arr[p]<nums2[q]?arr[p++]:nums2[q++]; } // k=k-1; while(p<m){ nums1[k++]=arr[p++]; } while(q<n){ nums1[k++]=nums2[q++]; } } // return nums1; } }