算法之寻找二维矩阵中的目标值

题目

 * 编写一个高效的算法来搜索 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。

分析

因为二维数组是有序的，所以不难想到使用二分法查找，方法很简单，这里贴出个人的初期代码。

/// <summary>
/// 在矩阵中搜索目标值
/// </summary>
/// <param name="matrix">二维矩阵</param>
/// <param name="target">目标值</param>
/// <returns>结果</returns>
public bool searchMatrix(int[,] matrix, int target)
{
    int row = matrix.GetLength(0) - 1;
    int col = matrix.GetLength(1) - 1;

    if (matrix == null || matrix.Length == 0 || col == 0) return false;

    for (int i = 0; i < row; i++)
    {
        int start = 0, end = col - 1, mid = 0;
        while (end - start > 1)
        {
            mid = (end + start) >> 1;
            if (matrix[i, mid] > target)
            {
                end = mid;
            }
            else if (matrix[i, mid] < target)
            {
                start = mid;
            }
            else
            {
                return true;
            }
        }
        if (matrix[i, start] == target || matrix[i, end] == target) return true;
    }
    return false;
}

代码不做过多解释，实在不理解的先去了解下二分法的具体过程。本函数有一处右移(>>)的操作，作用跟除以2差不多，不习惯的朋友可以把 >> 1改为 / 2。

验证

C#初始化二维数组的代码可以这样写：

int[,] matrix = new int[,]
{
	{ 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比较，然后根据比较结果再进行下一步比较不就ok了吗？

更优的代码

这里采用的是把矩阵左下角元素作为起始点，如果想选矩阵右上角元素作为起始点的朋友可以自行去写出这部分，方法都是类似的。

public bool SearchMatrix(int[,] matrix, int target)
{
    int row = matrix.GetLength(0) - 1;
    int col = 0;

    while (row >= 0 && col < matrix.GetLength(1))
    {
        if (matrix[row, col] > target) col++;
        else if (matrix[row, col] < target) row--;
        else return true;
    }
    return false;
}

相信经过这个题，大家一定也对算法产生了更深的兴趣，很喜欢的一句话分享给你们。
From zero to hero.