mirror of
https://github.com/TheAlgorithms/Java.git
synced 2025-07-27 22:43:30 +08:00
Add algorithm to search for an element in 2D array (#3306)
This commit is contained in:
@ -0,0 +1,124 @@
|
||||
package com.thealgorithms.searches;
|
||||
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||
|
||||
|
||||
public class RowColumnWiseSorted2dArrayBinarySearchTest {
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArrayBinarySearchTestMiddle() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 35;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {1,2};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(1, ans[0]);
|
||||
assertEquals(2, ans[1]);
|
||||
}
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArrayBinarySearchTestSide() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 48;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {2,3};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(2, ans[0]);
|
||||
assertEquals(3, ans[1]);
|
||||
}
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArray_BinarySearchTestUpper() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 20;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {0,1};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(0, ans[0]);
|
||||
assertEquals(1, ans[1]);
|
||||
}
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArray_BinarySearchTestUpperSide() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 40;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {0,3};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(0, ans[0]);
|
||||
assertEquals(3, ans[1]);
|
||||
}
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArray_BinarySearchTestLower() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 31;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {3,1};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(3, ans[0]);
|
||||
assertEquals(1, ans[1]);
|
||||
}
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArray_BinarySearchTestLowerSide() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 51;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {3,3};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(3, ans[0]);
|
||||
assertEquals(3, ans[1]);
|
||||
}
|
||||
|
||||
@Test
|
||||
// valid test case
|
||||
public void rowColumnSorted2dArray_BinarySearchTestNotFound() {
|
||||
int[][] arr = { {10, 20, 30, 40},
|
||||
{15, 25, 35, 45},
|
||||
{18, 28, 38, 48},
|
||||
{21, 31, 41, 51}
|
||||
};
|
||||
int target = 101;
|
||||
int[] ans = RowColumnWiseSorted2dArrayBinarySearch.search(arr, target);
|
||||
int[] expected = {-1,-1};
|
||||
System.out.println(Arrays.toString(ans));
|
||||
assertEquals(-1, ans[0]);
|
||||
assertEquals(-1, ans[1]);
|
||||
}
|
||||
|
||||
}
|
Reference in New Issue
Block a user