數據結構與算法《四》

一切推理都必須從觀察與實驗得來。 —— 伽利略·伽利雷

LeetCode: 搜索二維矩陣 II

編寫一個高效的算法來搜索 m x n 矩陣 matrix 中的一個目標值 target。該矩陣具備如下特性：

• 每行的元素從左到右升序排列。
• 每列的元素從上到下升序排列。

示例:
現有矩陣 matrix 以下：

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]

給定 target = 5，返回 true。

給定 target = 20，返回 false。

分析：

解法一：

由於矩陣每一行都是升序排列，因此能夠使用二分查找，對每行作二分查找。

Code:

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        for (int i = 0; i < matrix.length; ++i) {
            int left = 0;
            int right = matrix[i].length - 1;
            while (left <= right) {
                int mid = (left + right) / 2;
                if (matrix[i][mid] > target) {
                    right = mid - 1;
                } else if (matrix[i][mid] < target) {
                    left = mid + 1;
                } else {
                    return true;
                }
            }
        }
        return false;
    }
}

解法二：

既然每行的元素從左到右升序排列，每列的元素從上到下升序排列。則左下角的元素爲最小值，以此元素爲基準，小於目標值，則向右移動，大於目標值，則向上移動。

Code:

class Solution {
    public boolean searchMatrix(int[][] matrix, int target) {
        int row = matrix.length - 1;
        int col = 0;
        while (row >= 0 && col <= matrix[0].length - 1) {
            if (matrix[row][col] == target) {
                return true;
            } else if (matrix[row][col] > target) {
                row--;
            } else {
                col++;
            }
        }
        return false;
    }
}

按期分享大數據與知識圖譜相關知識點：