Leetcode 88.合併兩個有序陣列
阿新 • • 發佈:2018-12-22
題目:
給定兩個有序整數陣列 nums1 和 nums2,將 nums2 合併到 nums1 中,使得 num1 成為一個有序陣列。
說明:
- 初始化 nums1 和 nums2 的元素數量分別為 m 和 n。
- 你可以假設 nums1 有足夠的空間(空間大小大於或等於 m + n)來儲存 nums2 中的元素。
示例:
輸入: nums1 = [1,2,3,0,0,0], m = 3 nums2 = [2,5,6], n = 3 輸出: [1,2,2,3,5,6]
分析:
方法一:剛開始想建立一個新的陣列,長度等於m+n。之後比較兩個陣列的每個元素,將小的存入新的陣列,這種方法當其中一個數組遍歷完成時,另一個數組會有剩餘的元素,需要將其手動新增到後面的元素。
package LanQiao; import java.util.Arrays; public class Main { public static void main(String[] args) { int[] nums1 = { 1, 2, 3, 0, 0, 0 }; int[] nums2 = { 2, 5, 6 }; merge1(nums1, 3, nums2, 3); } public static void merge(int[] nums1, int m, int[] nums2, int n) { int[] num = new int[m + n]; int i = 0, j = 0, k = 0; while (i < m && j < n) { if (nums1[i] <= nums2[j]) num[k++] = nums1[i++]; else num[k++] = nums2[j++]; } System.out.println(i); System.out.println(j); while (i < m) num[k++] = nums1[i++]; while (j < n) num[k++] = nums2[j++]; for (int k1 = 0; i < num.length; k1++) { System.out.println(num[k1]); } } }
這種方法原理上沒有錯,但是在LeetCode上卻執行錯誤。是因為題目上要求將 nums2 合併到 nums1 中,而我是將其賦值給新的陣列。
方法二:將nums2 合併到 nums1 後面,之後使用排序方法即可。
package LanQiao; import java.util.Arrays; public class Main { public static void main(String[] args) { int[] nums1 = { 1, 2, 3, 0, 0, 0 }; int[] nums2 = { 2, 5, 6 }; merge1(nums1, 3, nums2, 3); } public static void merge1(int[] nums1, int m, int[] nums2, int n) { if (m <= 0) { nums1[0] = nums2[0]; } for (int i = 0; i < nums2.length; i++) { nums1[m + i] = nums2[i]; } Arrays.sort(nums1); for (int i = 0; i < nums1.length; i++) { System.out.println(nums1[i]); } } }
對陣列排序可以使用自己寫的氣泡排序,也可以使用Arrays類中封裝好的sort方法,並且這種方法在速度上也有較大的提升。
執行結果: