1. 程式人生 > >Leetcode 88.合併兩個有序陣列

Leetcode 88.合併兩個有序陣列

題目:

給定兩個有序整數陣列 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方法,並且這種方法在速度上也有較大的提升。

執行結果: