LeetCode-in-All

74. Search a 2D Matrix

Medium

Write an efficient algorithm that searches for a value target in an m x n integer matrix matrix. This matrix has the following properties:

• Integers in each row are sorted from left to right.
• The first integer of each row is greater than the last integer of the previous row.

Example 1:

Input: matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 3

Output: true

Example 2:

Input: matrix = [[1,3,5,7],[10,11,16,20],[23,30,34,60]], target = 13

Output: false

Constraints:

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 100
• -104 <= matrix[i][j], target <= 104

Solution

#include <stdio.h>
#include <stdbool.h>

bool searchMatrix(int** matrix, int matrixSize, int* matrixColSize, int target) {
    int endRow = matrixSize;
    int endCol = *matrixColSize;
    int targetRow = 0;
    bool result = false;

    // Step 1: Find the row that may contain the target
    for (int i = 0; i < endRow; i++) {
        if (matrix[i][endCol - 1] >= target) {
            targetRow = i;
            break;
        }
    }

    // Step 2: Search within the target row for the target value
    for (int i = 0; i < endCol; i++) {
        if (matrix[targetRow][i] == target) {
            result = true;
            break;
        }
    }

    return result;
}

This site is open source. Improve this page.