240. Search a 2D Matrix II
Problem Description
The problem is to create an algorithm that can efficiently search for a specific value, called target
, within a 2-dimensional matrix. The dimensions of the matrix are m x n
, which means it has m
rows and n
columns. The matrix is not just any random assortment of integers—it has two key properties that can be leveraged to make searching efficient:
- Each row of the matrix is sorted in ascending order from left to right.
- Each column of the matrix is sorted in ascending order from top to bottom.
Given these sorted properties of the matrix, the search should be done in a way that is more optimized than a brute-force approach which would check every element.
Intuition
To intuitively understand the solution, we should recognize that because of the row and column properties, the matrix resembles a 2D binary search problem. However, instead of splitting our search space in half each time as we would in a conventional binary search, we can make an observation:
- If we start in the bottom-left corner of the matrix (or the top-right), we find ourselves at an interesting position: moving up decreases the value (since columns are sorted in ascending order from top to bottom), and moving right increases the value (since rows are sorted in ascending order from left to right).
This starting point gives us a "staircase" pattern to follow based on comparisons:
- If the
target
is greater than the value at our current position, we know it can't be in the current row (to the left), so we move to the right (increase the column index). - If the
target
is less than the value at our current position, we know it can't be in the current column (below), so we move up (decrease the row index).
By doing this, we are eliminating either a row or a column at each step, leveraging the matrix's properties to find the target
or conclude it's not there. We keep this up until we find the target
or exhaust all our moves (when we move out of the bounds of the matrix), in which case the target
is not present. This is why the while loop condition in the code checks if i >= 0 and j < n
.
The process is very much like tracing a path through the matrix that "zig-zags" closer and closer to the value if it's present. The solution here effectively combines aspects of both binary search and linear search but applied in a 2-dimensional space.
Learn more about Binary Search and Divide and Conquer patterns.
Solution Approach
The solution provided is a direct implementation of the intuitive strategy discussed previously. Here's how the solution is implemented:
-
We declare a
Solution
class with a methodsearchMatrix
that accepts a 2D matrix and thetarget
value as parameters. -
Inside the
searchMatrix
method, we start by getting the dimensions of the matrix:m
for the number of rows andn
for the number of columns, which will be used for boundary checking during the search. -
We initiate two pointers,
i
andj
, which will traverse the matrix.i
is initialized tom - 1
, which means it starts from the last row (bottom row), andj
is initialized to0
, which is the first column (leftmost column). This represents the bottom-left corner of the matrix. -
The search begins and continues as long as our pointers are within the bounds of the matrix. The
while
loop condition makes sure thati
is never less than0
(which would mean we've moved above the first row) andj
is less thann
(to ensure we don't move beyond the last column to the right). -
Within the loop, there are three cases to consider:
- If the element at the current position
matrix[i][j]
equals thetarget
, then our search is successful, and we returnTrue
. - If the element at
matrix[i][j]
is greater than thetarget
, we must move up (decrease the value ofi
) to find a smaller element. - Conversely, if
matrix[i][j]
is less than thetarget
, we must move right (increase the value ofj
) in hopes of finding a larger element.
- If the element at the current position
-
If we exit the loop without returning
True
, it means we have exhausted all possible positions in the matrix without finding thetarget
, and thus we returnFalse
.
This approach effectively traverses the matrix in a manner such that with each comparison, a decision can be made to eliminate either an entire row or an entire column, significantly reducing the search space and making the algorithm efficient. The worst case scenario would be traversing from the bottom-left corner to the top-right corner, which gives us a time complexity of O(m + n), where m
is the number of rows and n
is the number of columns.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Let's walk through a small example to illustrate the solution approach. Consider the following 3x4 matrix and a target value to search for:
[ [1, 4, 7, 11], [2, 5, 8, 12], [3, 6, 9, 16] ]
And let's say we are searching for the target value 5
.
Using the solution approach, we start at the bottom-left corner of the matrix. This means our starting position is at the last row, first column: matrix[2][0]
, which is 3
.
Now, we compare the target value 5
with the value at our current position:
-
target
(5) >matrix[2][0]
(3) so the target can't be in the current row because all values to the left are smaller. We move right to increase the value (incrementj
): now we are atmatrix[2][1]
, which is6
. -
target
(5) <matrix[2][1]
(6) so the target can't be in the current column because all values below are larger. We move up to decrease the value (decrementi
): now we are atmatrix[1][1]
, which is5
.
At this point, matrix[1][1]
equals the target value 5
, so our search is successful, and we return True
.
This approach avoided checking every single element in the matrix, instead, by leveraging the sorted nature of the rows and columns, it quickly hones in on the target with a clear strategy. Even if the target number was not present, the process would eventually move out of the matrix bounds, at which point we would return False
, signifying that the target is not found.
Solution Implementation
1class Solution:
2 def searchMatrix(self, matrix, target):
3 # Rows and columns in the matrix
4 num_rows, num_cols = len(matrix), len(matrix[0])
5
6 # Start from the bottom left corner of the matrix
7 row_index, col_index = num_rows - 1, 0
8
9 # Loop until we have a valid position within the matrix bounds
10 while row_index >= 0 and col_index < num_cols:
11 # Check if the current element is the target
12 if matrix[row_index][col_index] == target:
13 return True
14
15 # If current element is larger than target, move up to reduce value
16 if matrix[row_index][col_index] > target:
17 row_index -= 1
18 # If current element is smaller than target, move right to increase value
19 else:
20 col_index += 1
21
22 # Return False if we haven't returned True by this point
23 return False
24
1class Solution {
2
3 /**
4 * Searches for a target value in a 2D matrix.
5 * The matrix has the following properties:
6 * - Integers in each row are sorted in ascending from left to right.
7 * - The first integer of each row is greater than the last integer of the previous row.
8 *
9 * @param matrix 2D matrix of integers
10 * @param target The integer value to search for
11 * @return boolean indicating whether the target is found
12 */
13 public boolean searchMatrix(int[][] matrix, int target) {
14 // Get the number of rows and columns in the matrix
15 int rowCount = matrix.length;
16 int colCount = matrix[0].length;
17
18 // Start from the bottom-left corner of the matrix
19 int row = rowCount - 1;
20 int col = 0;
21
22 // Perform a staircase search
23 while (row >= 0 && col < colCount) {
24 if (matrix[row][col] == target) {
25 // Target is found at the current position
26 return true;
27 }
28 if (matrix[row][col] > target) {
29 // Target is less than the current element, move up
30 row--;
31 } else {
32 // Target is greater than the current element, move right
33 col++;
34 }
35 }
36
37 // Target was not found in the matrix
38 return false;
39 }
40}
41
1#include <vector>
2using namespace std;
3
4class Solution {
5public:
6 // Searches for a target value within a 2D matrix. This matrix has the following properties:
7 // Integers in each row are sorted in ascending from left to right.
8 // Integers in each column are sorted in ascending from top to bottom.
9 //
10 // @param matrix The matrix of integers.
11 // @param target The target integer to find.
12 // @return True if target is found, false otherwise.
13 bool searchMatrix(vector<vector<int>>& matrix, int target) {
14 // Get the number of rows.
15 int rows = matrix.size();
16 // Get the number of columns.
17 int columns = matrix[0].size();
18
19 // Start from the bottom-left corner of the matrix.
20 int currentRow = rows - 1;
21 int currentColumn = 0;
22
23 // While the position is within the bounds of the matrix...
24 while (currentRow >= 0 && currentColumn < columns) {
25 // If the current element is the target, return true.
26 if (matrix[currentRow][currentColumn] == target) return true;
27
28 // If the current element is larger than the target, move up one row.
29 if (matrix[currentRow][currentColumn] > target) {
30 --currentRow;
31 }
32 // If the current element is smaller than the target, move right one column.
33 else {
34 ++currentColumn;
35 }
36 }
37
38 // If the target is not found, return false.
39 return false;
40 }
41};
42
1/**
2 * Searches for a target value in a matrix.
3 * This matrix has the following properties:
4 * 1. Integers in each row are sorted from left to right.
5 * 2. The first integer of each row is greater than the last integer of the previous row.
6 * @param matrix A 2D array of numbers representing the matrix.
7 * @param target The number to search for in the matrix.
8 * @return A boolean indicating whether the target exists in the matrix.
9 */
10function searchMatrix(matrix: number[][], target: number): boolean {
11 // Get the number of rows (m) and columns (n) in the matrix
12 let rowCount = matrix.length,
13 columnCount = matrix[0].length;
14
15 // Start our search from the bottom-left corner of the matrix
16 let currentRow = rowCount - 1,
17 currentColumn = 0;
18
19 // Continue the search while we're within the bounds of the matrix
20 while (currentRow >= 0 && currentColumn < columnCount) {
21 // Retrieve the current element to compare with the target
22 let currentElement = matrix[currentRow][currentColumn];
23
24 // Check if the current element matches the target
25 if (currentElement === target) return true;
26
27 // If the current element is greater than the target,
28 // move up to the previous row since all values in the current
29 // row will be too large given the matrix's sorted properties
30 if (currentElement > target) {
31 --currentRow;
32 } else {
33 // If the current element is less than the target,
34 // move right to the next column since all values in previous
35 // columns will be too small given the matrix's sorted properties
36 ++currentColumn;
37 }
38 }
39
40 // If we've exited the while loop, the target is not present in the matrix
41 return false;
42}
43
Time and Space Complexity
Time Complexity
The time complexity of the search algorithm is O(m + n)
, where m
is the number of rows and n
is the number of columns in the matrix. This is because the algorithm starts from the bottom-left corner of the matrix and moves either up (i
decreases) or right (j
increases) at each step. At most, it will move m
steps upwards and n
steps to the right before it either finds the target or reaches the top-right corner, thus completing the search.
Space Complexity
The space complexity of the algorithm is O(1)
. This is because the algorithm uses a fixed amount of extra space (variables i
and j
) regardless of the size of the input matrix. It doesn't require any additional data structures that grow with the input size.
Learn more about how to find time and space complexity quickly using problem constraints.
Which algorithm should you use to find a node that is close to the root of the tree?
Recommended Readings
Binary Search Speedrun For each of the Speedrun questions you will be given a binary search related problem and a corresponding multiple choice question The multiple choice questions are related to the techniques and template s introduced in the binary search section It's recommended that you have gone through at
https algomonster s3 us east 2 amazonaws com cover_photos dnd svg Divide and Conquer The main idea of divide and conquer is to split the main problem into two smaller components assuming that each one of the components had already been solved recursively and then try to solve the bigger
LeetCode Patterns Your Personal Dijkstra's Algorithm to Landing Your Dream Job The goal of AlgoMonster is to help you get a job in the shortest amount of time possible in a data driven way We compiled datasets of tech interview problems and broke them down by patterns This way we
Want a Structured Path to Master System Design Too? Don’t Miss This!