4. 兩個排序陣列的中位數---LeetCode
阿新 • • 發佈:2018-12-22
題目描述:
給定兩個大小為 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
程式碼如下:
#define _CRT_SECURE_NO_DEPRECATE 1 #include "stdio.h" #include<stdlib.h> double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) { double ret; if (nums1Size == 0 && nums2Size != 0){ return ret = (double)(nums2[(nums2Size - 1) / 2] + nums2[nums2Size / 2]) / 2; } if (nums2Size == 0 && nums1Size != 0){ return ret = (double)(nums1[(nums1Size - 1) / 2] + nums1[nums1Size / 2]) / 2; } int k = 0; int i = 0; int j = 0; int* nums = (int *)malloc(sizeof(int)*(nums1Size+nums2Size)); int temp; for (k = 0; i < nums1Size && j < nums2Size && (k < (nums1Size + nums2Size)); k++){ if (i < nums1Size && nums1[i] <= nums2[j]){ temp = nums1[i]; i++; } else{ temp = nums2[j]; j++; } nums[k] = temp; } if (i < nums1Size && j >= nums2Size){ for (i; i < nums1Size; i++){ nums[k++] = nums1[i]; } } if (j < nums2Size && i >= nums1Size){ for (j; j < nums2Size; j++){ nums[k++] = nums2[j]; } } return ret = (double)(nums[(k - 1) / 2] + nums[k / 2]) / 2; } int main() { int nums1[] = {1,3}; int nums2[] = {2}; int size1 = sizeof(nums1) / sizeof(int); int size2 = sizeof(nums2) / sizeof(int); double ret = findMedianSortedArrays(nums1, size1, nums2, size2); printf("%1f\n", ret); return 0; }