1. 程式人生 > >LeetCode第4題

LeetCode第4題

給定兩個大小為 m 和 n 的有序陣列 nums1 和 nums2

請你找出這兩個有序陣列的中位數,並且要求演算法的時間複雜度為 O(log(m + n))。

你可以假設 nums1 和 nums2 不會同時為空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

則中位數是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

則中位數是 (2 + 3)/2 = 2.5

思路:我自己的思路是比較笨的方法。

1.定義一個List集合,將nums1和nums2中的元素新增到List集合中。

2.利用集合操作類Collections來將集合排序。

3.如果集合中的元素個數是奇數的話,則輸出一箇中位數;如果是偶數的話,則輸出最中間的兩個數的和的平均數。

public static double findMedianSortedArrays(int[] nums1,int[] nums2){
		double result=0.0;
		List<Integer> list=new ArrayList<Integer>();
		for(int i=0;i<nums1.length;i++){
			list.add(nums1[i]);
		}
		for(int j=0;j<nums2.length;j++){
			list.add(nums2[j]);
		}
		Collections.sort(list);
		if(list.size()%2==0){
			int index=list.size()/2;
			int result1=list.get(index)+list.get(index-1);
			result=result1/2;
			return result;
		}
		if(list.size()%2==1){
			int index2=list.size()/2;
			return list.get(index2);
		}
		return result;
		
	}

這個題的思路很簡單,如果大家有更好的想法的話,可以留言評論奧!!