**
Question:
There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
**
My code:
public class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
if (nums1 == null || nums2 == null)
return 0;
if (nums1.length == 0 && nums2.length == 0)
return 0;
int m = nums1.length;
int n = nums2.length;
if (m == 0)
if (n % 2 == 1)
return nums2[n / 2];
else
return ((double)nums2[n / 2] + (double)nums2[n / 2 - 1]) / 2.0;
else if (n == 0)
if (m % 2 == 1)
return nums1[m / 2];
else
return ((double)nums1[m / 2] + (double)nums1[m / 2 - 1]) / 2.0;
int indexOfNums1 = 0;
int indexOfNums2 = 0;
boolean isOutOfRange = false;
int[] mergeArray = new int[m + n];
for (int i = 0; i < m + n; i++) {
if (indexOfNums1 < m && (nums1[indexOfNums1] < nums2[indexOfNums2] || isOutOfRange)) {
mergeArray[i] = nums1[indexOfNums1];
indexOfNums1++;
}
else {
mergeArray[i] = nums2[indexOfNums2];
if (indexOfNums2 < n - 1)
indexOfNums2++;
else
isOutOfRange = true;
}
}
if ((m + n) % 2 == 1)
return mergeArray[(m + n) / 2];
else
return ((double)mergeArray[(m + n) / 2] + (double)mergeArray[(m + n) / 2 - 1]) / 2.0;
}
public static void main(String[] args) {
Solution test = new Solution();
int[] a = {100001};
int[] b = {100000};
System.out.println(test.findMedianSortedArrays(a, b));
}
}
My test result:
这次的难度是hard,但是我一看到就有了思路,然后花了十分钟写了出来。
这不就是归并排序(Merge Sort)的简单版本么。。。
给两个已排序好的数列,然后将它们合并,获得一个新的数列
来,复习下归并排序。
我自己创造的一个图书。
. . . . . . . . 一个array,用归并排序来处理
. . . . | . . . . 分成两个array,分别进行归并排序
. . | . . | . . | . . 对于每个子array,再次分成两个array,分别进行归并排序。
此时每个部分只有两个数字了, lo指向第一个,头。hi指向最后一个,也是第二个,尾。
于是进行简单的排序。完成了。退回到上步。
. . | . . | . . | . . 此时的子array都是已经排序好了的。于是将 1,2子array归并,3,4子array归并。
. . . . | . . . . 此时的两个子array也是已经排序好了的。于是将它们归并,得到最后的排序结果。
. . . . . . . .
然后归并排序的核心就是这次作业写的。
两个array,然后我先创建一个array长度是他们的长度和。
然后通过移动指针,将两个数组对应位置较小的拷贝进入mergeArray,然后指针再往前一格。这样,扫描(m + n)次就能完成了。
然后处理一些数组越界问题。就差不多。复杂度是 N * Log(N)
**
总结:
归并排序。
数组越界。
**
最近两次作业都是十几分钟写出来了。。。是我太强了吗?
我不觉得。我觉得是评测系统有问题。这两道题目正常人应该都半小时内就能做出来吧。
最近听了荔枝FM的一个广播。额,首先声明,我绝对不是做广告的人。
很感动。然后那么一句话。
**
其实我也是一个普通人啊。
**
打到我内心了。
小说世界里, 泰广,米可,最后能破镜重圆。
现实世界里,会有这般失而复得的事吗?
所以,确定你想要的,然后坚持下去。
读者们肯定看不懂我说的这些话,但你呢,应该看得懂吧。
我愿意为你改变很多,放弃很多。
你也愿意为我忍耐很多,担心很多。
异地异国实在是太难熬,但熬出头了呢。
如果因为那么一件件的事,咱们最终放弃。不知道你会怎么想。
我的话,一定会有新的女朋友,家庭。
但这份对你的遗憾,会陪伴我,直到进入棺材。
Anyway, Good luck, Richardo!
My code:
public class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
if (nums1 == null || nums2 == null) {
return -1;
}
else if (nums2.length < nums1.length) {
return findMedianSortedArrays(nums2, nums1);
}
else if (nums2.length == 0) {
return -1;
}
int m = nums1.length;
int n = nums2.length;
int half = (m + n + 1) / 2;
int imin = 0;
int imax = m;
while (imin <= imax) {
int i = (imax + imin) / 2;
int j = half - i;
if (i - 1 >= 0 && j < n && nums1[i - 1] > nums2[j]) {
imax = i - 1;
}
else if (j - 1 >= 0 && i < m && nums2[j - 1] > nums1[i]) {
imin = i + 1;
}
else {
int left = 0;
if (i == 0) {
left = nums2[j - 1];
}
else if (j == 0) {
left = nums1[i - 1];
}
else {
left = Math.max(nums1[i - 1], nums2[j - 1]);
}
if ((m + n) % 2 == 1) {
return left;
}
int right = 0;
if (i == m) {
right = nums2[j];
}
else if (j == n) {
right = nums1[i];
}
else {
right = Math.min(nums1[i], nums2[j]);
}
return (left + right) / 2.0;
}
}
return -1;
}
public static void main(String[] args) {
Solution test = new Solution();
int[] nums1 = new int[]{1,2};
int[] nums2 = new int[]{3,4,5,6};
double ret = test.findMedianSortedArrays(nums1, nums2);
System.out.println(ret);
}
}
reference:
https://discuss.leetcode.com/topic/4996/share-my-o-log-min-m-n-solution-with-explanation/2
这道题目实在是太恶心了。弄了一个多小时。
corner case 太多了。。。
看下讲解,思路清晰了很多。但是还是调了很久很久。
最大的问题是,得把短的作为切割的第一数组,即, i
因为如果切长的,作为i
那么 j很有可能变成负数,会出现错误。
如果切短的,就不会用这个问题。
还有个注意的地方就是:
half = (m + n + 1) / 2;
这样保证,如果 m + n 是偶数,
那么左右部分个数相等。
如果是奇数,那么左边会多一个。
而且imax = m 而不是 m - 1,真的求中点。
如果m是奇数,就是正中点。如果m是偶数,是中间两个数右边那个。
这些正好都是我们需要的特性。
实在是太恶心了。
加油!
Anyway, Good luck, Richardo! -- 09/01/2016