Skip to content

Commit 5c79e5d

Browse files
authored
Add MedianOfTwoSortedArrays algorithm (#5554)
1 parent 403649d commit 5c79e5d

File tree

3 files changed

+99
-0
lines changed

3 files changed

+99
-0
lines changed

DIRECTORY.md

+5
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
* [IsPowerTwo](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/IsPowerTwo.java)
3232
* [LowestSetBit](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/LowestSetBit.java)
3333
* [NonRepeatingNumberFinder](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/NonRepeatingNumberFinder.java)
34+
* [NumberAppearingOddTimes](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/NumberAppearingOddTimes.java)
3435
* [NumbersDifferentSigns](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/NumbersDifferentSigns.java)
3536
* [ReverseBits](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/ReverseBits.java)
3637
* [SingleBitOperations](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/bitmanipulation/SingleBitOperations.java)
@@ -228,6 +229,7 @@
228229
* divideandconquer
229230
* [BinaryExponentiation](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/divideandconquer/BinaryExponentiation.java)
230231
* [ClosestPair](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/divideandconquer/ClosestPair.java)
232+
* [MedianOfTwoSortedArrays](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/divideandconquer/MedianOfTwoSortedArrays.java)
231233
* [SkylineAlgorithm](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/divideandconquer/SkylineAlgorithm.java)
232234
* [StrassenMatrixMultiplication](https://github.com/TheAlgorithms/Java/blob/master/src/main/java/com/thealgorithms/divideandconquer/StrassenMatrixMultiplication.java)
233235
* dynamicprogramming
@@ -638,6 +640,7 @@
638640
* [IsPowerTwoTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/IsPowerTwoTest.java)
639641
* [LowestSetBitTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/LowestSetBitTest.java)
640642
* [NonRepeatingNumberFinderTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/NonRepeatingNumberFinderTest.java)
643+
* [NumberAppearingOddTimesTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/NumberAppearingOddTimesTest.java)
641644
* [NumbersDifferentSignsTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/NumbersDifferentSignsTest.java)
642645
* [ReverseBitsTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/ReverseBitsTest.java)
643646
* [SingleBitOperationsTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/bitmanipulation/SingleBitOperationsTest.java)
@@ -729,6 +732,7 @@
729732
* [LeftistHeapTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/datastructures/heaps/LeftistHeapTest.java)
730733
* lists
731734
* [CircleLinkedListTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/datastructures/lists/CircleLinkedListTest.java)
735+
* [CreateAndDetectLoopTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/datastructures/lists/CreateAndDetectLoopTest.java)
732736
* [QuickSortLinkedListTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/datastructures/lists/QuickSortLinkedListTest.java)
733737
* [ReverseKGroupTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/datastructures/lists/ReverseKGroupTest.java)
734738
* [RotateSinglyLinkedListsTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/datastructures/lists/RotateSinglyLinkedListsTest.java)
@@ -773,6 +777,7 @@
773777
* divideandconquer
774778
* [BinaryExponentiationTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/divideandconquer/BinaryExponentiationTest.java)
775779
* [ClosestPairTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/divideandconquer/ClosestPairTest.java)
780+
* [MedianOfTwoSortedArraysTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/divideandconquer/MedianOfTwoSortedArraysTest.java)
776781
* [SkylineAlgorithmTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/divideandconquer/SkylineAlgorithmTest.java)
777782
* [StrassenMatrixMultiplicationTest](https://github.com/TheAlgorithms/Java/blob/master/src/test/java/com/thealgorithms/divideandconquer/StrassenMatrixMultiplicationTest.java)
778783
* dynamicprogramming
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,53 @@
1+
package com.thealgorithms.divideandconquer;
2+
3+
public final class MedianOfTwoSortedArrays {
4+
5+
private MedianOfTwoSortedArrays() {
6+
}
7+
8+
/**
9+
* Finds the median of two sorted arrays in logarithmic time.
10+
*
11+
* @param nums1 the first sorted array
12+
* @param nums2 the second sorted array
13+
* @return the median of the combined sorted array
14+
* @throws IllegalArgumentException if the input arrays are not sorted
15+
*/
16+
public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
17+
if (nums1.length > nums2.length) {
18+
return findMedianSortedArrays(nums2, nums1); // Ensure nums1 is the smaller array
19+
}
20+
21+
int m = nums1.length;
22+
int n = nums2.length;
23+
int low = 0;
24+
int high = m;
25+
while (low <= high) {
26+
int partition1 = (low + high) / 2; // Partition in the first array
27+
int partition2 = (m + n + 1) / 2 - partition1; // Partition in the second array
28+
29+
int maxLeft1 = (partition1 == 0) ? Integer.MIN_VALUE : nums1[partition1 - 1];
30+
int minRight1 = (partition1 == m) ? Integer.MAX_VALUE : nums1[partition1];
31+
int maxLeft2 = (partition2 == 0) ? Integer.MIN_VALUE : nums2[partition2 - 1];
32+
int minRight2 = (partition2 == n) ? Integer.MAX_VALUE : nums2[partition2];
33+
34+
// Check if partition is valid
35+
if (maxLeft1 <= minRight2 && maxLeft2 <= minRight1) {
36+
// If combined array length is odd
37+
if (((m + n) & 1) == 1) {
38+
return Math.max(maxLeft1, maxLeft2);
39+
}
40+
// If combined array length is even
41+
else {
42+
return (Math.max(maxLeft1, maxLeft2) + Math.min(minRight1, minRight2)) / 2.0;
43+
}
44+
} else if (maxLeft1 > minRight2) {
45+
high = partition1 - 1;
46+
} else {
47+
low = partition1 + 1;
48+
}
49+
}
50+
51+
throw new IllegalArgumentException("Input arrays are not sorted");
52+
}
53+
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
package com.thealgorithms.divideandconquer;
2+
3+
import static org.junit.jupiter.api.Assertions.assertEquals;
4+
5+
import java.util.stream.Stream;
6+
import org.junit.jupiter.params.ParameterizedTest;
7+
import org.junit.jupiter.params.provider.Arguments;
8+
import org.junit.jupiter.params.provider.MethodSource;
9+
10+
public class MedianOfTwoSortedArraysTest {
11+
12+
@ParameterizedTest
13+
@MethodSource("provideTestCases")
14+
void testFindMedianSortedArrays(int[] nums1, int[] nums2, double expectedMedian) {
15+
assertEquals(expectedMedian, MedianOfTwoSortedArrays.findMedianSortedArrays(nums1, nums2));
16+
}
17+
18+
private static Stream<Arguments> provideTestCases() {
19+
return Stream.of(
20+
// Test case 1: Arrays of equal length
21+
Arguments.of(new int[] {1, 3}, new int[] {2, 4}, 2.5),
22+
23+
// Test case 2: Arrays of different lengths
24+
Arguments.of(new int[] {1, 3}, new int[] {2}, 2.0),
25+
26+
// Test case 3: Arrays with even total length
27+
Arguments.of(new int[] {1, 2, 8}, new int[] {3, 4, 5, 6, 7}, 4.5),
28+
29+
// Test case 4: Arrays with odd total length
30+
Arguments.of(new int[] {1, 2, 8}, new int[] {3, 4, 5}, 3.5),
31+
32+
// Test case 5: Single element arrays
33+
Arguments.of(new int[] {1}, new int[] {3}, 2.0),
34+
35+
// Test case 6: Empty arrays
36+
Arguments.of(new int[] {}, new int[] {0}, 0.0),
37+
38+
// Test case 7: Same element arrays
39+
Arguments.of(new int[] {2, 2, 2}, new int[] {2, 2, 2}, 2.0));
40+
}
41+
}

0 commit comments

Comments
 (0)