LeetCode第4題
阿新 • • 發佈:2018-12-27
給定兩個大小為 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; }
這個題的思路很簡單,如果大家有更好的想法的話,可以留言評論奧!!