88. 合併兩個有序陣列
知識點:陣列;排序;雙指標;
題目描述
給你兩個按 非遞減順序 排列的整數陣列 nums1 和 nums2,另有兩個整數 m 和 n ,分別表示 nums1 和 nums2 中的元素數目。
請你 合併 nums2 到 nums1 中,使合併後的陣列同樣按 非遞減順序 排列。
注意:最終,合併後陣列不應由函式返回,而是儲存在陣列 nums1 中。為了應對這種情況,nums1 的初始長度為 m + n,其中前 m 個元素表示應合併的元素,後 n 個元素為 0 ,應忽略。nums2 的長度為 n 。
示例
輸入: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 中的元素。
輸入:nums1 = [1], m = 1, nums2 = [], n = 0
輸出:[1]
解釋:需要合併 [1] 和 [] 。
合併結果是 [1] 。
解法一:排序
=這道題很簡單,可以直接把num2合併到num1上,也就是將兩個陣列合並,然後去排序;
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
for(int i = m; i < m+n; i++){
nums1[i] = nums2[i-m];
}
for(int i = 0; i < m+n; i++){
for(int j = 0; j < m+n-i-1; j++){
if(nums1[j] > nums1[j+1]){
int temp = nums1[j];
nums1[j] = nums1[j+1];
nums1[j+1] = temp;
}
}
}
}
}
解法二:雙指標
上面沒有用到一個條件:兩個陣列都是排序好的,所以可以定義兩個指標,對於兩個陣列都從頭到尾遍歷。經典的以空間換時間;
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
int[] arr = new int[m+n];
int index1 = 0, index2 = 0;
int index = 0;
while(index1 < m || index2 < n){
if(index1 == m){
arr[index++] = nums2[index2++];
}else if(index2 == n){
arr[index++] = nums1[index1++];
}else if(nums1[index1] <= nums2[index2]){
arr[index++] = nums1[index1++];
}else if(nums1[index1] > nums2[index2]){
arr[index++] = nums2[index2++];
}
}
for(int i = 0; i < m+n; i++){
nums1[i] = arr[i];
}
}
}