-
Notifications
You must be signed in to change notification settings - Fork 119
/
Copy pathMedian of Two Sorted Arrays.java
55 lines (51 loc) · 1.41 KB
/
Median of Two Sorted Arrays.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// Runtime: 5 ms (Top 46.02%) | Memory: 50.6 MB (Top 18.35%)
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
if(nums1.length==1 && nums2.length==1) return (double)(nums1[0]+nums2[0])/2.0;
int i = 0;
int j = 0;
int k = 0;
int nums[] = new int[nums1.length + nums2.length];
if(nums1.length !=0 && nums2.length != 0){
while(i < nums1.length && j < nums2.length){
if(nums1[i] < nums2[j]){
nums[k] = nums1[i];
i++;
k++;
}else{
nums[k] = nums2[j];
j++;
k++;
}
}
while(i < nums1.length){
nums[k] = nums1[i];
i++;
k++;
}
while(j < nums2.length){
nums[k] = nums2[j];
j++;
k++;
}
}
if(nums1.length==0){
for(int h: nums2){
nums[k] = h;
k++;
}
}
if(nums2.length==0){
for(int d : nums1){
nums[k] = d;
k++;
}
}
int mid = (nums.length / 2);
if (nums.length % 2 == 0) {
return ((double) nums[mid] + (double) nums[mid - 1]) / 2.0;
} else {
return nums[mid];
}
}
}