二维数组中的查找（C++和Python实现）

（说明：本博客中的题目、题目详细说明及参考代码均摘自 “何海涛《剑指Offer：名企面试官精讲典型编程题》2012年”）

题目

　　在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断是否含有该整数。

　　进一步的详细说明：

　　下图中的二维数组就是每行、每列都递增排序。如果在这个数组中查找数字 7，则返回 true；如果查找数字 5，由于数组不含有该数字，则返回 false。

算法设计思想

　　对于类似上图中的二维数组，选择从右上方（数字 9）或 左下方（数字 6）开始遍历，这样可以保证向单一方向缩小搜索范围；而从左上方（数字 1）或右下方（数字 15）开始遍历，可以朝两个方向进行进一步搜索，从而导致无法缩小范围。

以查找数字 5 为例，
　　若从左下方（数字 6）开始遍历，则由于数字 5 < 6 ，根据二维数组沿行从左到右递增，沿列从上到下递增，数字 5 可能位于数字 6 的上方或左侧，而数字 6 的左侧无元素，因此数字 5 只能位于数字 6 的上方，从而缩小了搜索范围；
　　若从左上方（数字 1）开始遍历，则由于数字 5 > 1，根据二维数组沿行从左到右递增，沿列从上到下递增，数字 5 可能位于数字 1 的下方或右侧，而数字 1的下方和右侧均有元素，因此无法缩小搜索范围。

　　以上说明，找到一个好的起点很重要，其会直接影响算法的可行性。

C++ 实现

#include <iostream>

using namespace std;

// Try to find 'target' from 'matrix' with 'rows' rows and 'cols' columns,
// If found, return true; else return false
bool Find(int* matrix, int rows, int cols, int target)
{
    bool found = false;

    if (!matrix || rows <  || cols < )  // 易漏点
        return found;
    int i = ;
    int j = cols-;   // Note: the index of the last element in each row is cols-1. // 易错点
    while ((i < rows) && (j >= ))
    {
        int current = *(matrix+i*cols+j);

        if (target < current)
            j--;
        else if (target > current)
            i++;
        else {  // target == current
            found = true;
            break;
        }
    }

    return found;
}

// Display the result whether the target is found.
void display(int* matrix, int rows, int cols, int target)
{
    cout << "Target " << target << " is ";
    if (Find(matrix, rows, cols, target))
        cout << "";
    else
        cout << "not ";
    cout << "found." << endl;
}

void unitest()
{
    int mat[] = {, ,  ,  ,
                 , ,  , ,
                 , , , ,
                 , , , };
    // Try to find digits 7 and 5, then display the result
    display(mat, , , );
    display(mat, , , );
}

int main()
{
    unitest();
    return ;
}

Python 实现

#!/usr/bin/python
# -*- coding: utf8 -*-

# Try to find 'target' from 'matrix' with 'rows' rows and 'cols' columns,
# If found, return true; else return false
def find_target(matrix, rows, cols, target):
    found = False
    if not matrix or rows < 0 or cols < 0:  # 易漏点
        return found
    i, j = 0, cols-1  # 易错点
    while i < rows and j >= 0:
        current = matrix[i*cols+j]
        if target < current:
            j -= 1
        elif target > current:
            i += 1
        else:  # target == current
            found = True
            break

    return found

def unitest():
    mat = [1, 2,  8,  9,
           2, 4,  9, 12,
           4, 7, 10, 13,
           6, 8, 11, 15]
    # Try to find digits 7 and 5, then display the result
    for target in [7, 5]:
        print "Target %d is %s found." % (target, "" if find_target(mat, 4, 4, target) else "not")

if __name__ == "__main__":
    unitest()

参考代码

1. targetver.h

#pragma once

// The following macros define the minimum required platform.  The minimum required platform
// is the earliest version of Windows, Internet Explorer etc. that has the necessary features to run
// your application.  The macros work by enabling all features available on platform versions up to and
// including the version specified.

// Modify the following defines if you have to target a platform prior to the ones specified below.
// Refer to MSDN for the latest info on corresponding values for different platforms.
#ifndef _WIN32_WINNT            // Specifies that the minimum required platform is Windows Vista.
#define _WIN32_WINNT 0x0600     // Change this to the appropriate value to target other versions of Windows.
#endif

2. stdafx.h

// stdafx.h : include file for standard system include files,
// or project specific include files that are used frequently, but
// are changed infrequently
//

#pragma once

#include "targetver.h"

#include <stdio.h>
#include <tchar.h>

// TODO: reference additional headers your program requires here

3. stdafx.cpp

// stdafx.cpp : source file that includes just the standard includes
// FindInPartiallySortedMatrix.pch will be the pre-compiled header
// stdafx.obj will contain the pre-compiled type information

#include "stdafx.h"

// TODO: reference any additional headers you need in STDAFX.H
// and not in this file

4. FindInPartiallySortedMatrix.cpp

// FindInPartiallySortedMatrix.cpp : Defines the entry point for the console application.
//

// 《剑指Offer——名企面试官精讲典型编程题》代码
// 著作权所有者：何海涛

#include "stdafx.h"

// 二维数组matrix中，每一行都从左到右递增排序，
// 每一列都从上到下递增排序
bool Find(int* matrix, int rows, int columns, int number)
{
    bool found = false;

    if(matrix != NULL && rows >  && columns > )
    {
        int row = ;
        int column = columns - ;
        while(row < rows && column >=)
        {
            if(matrix[row * columns + column] == number)
            {
                found = true;
                break;
            }
            else if(matrix[row * columns + column] > number)
                -- column;
            else
                ++ row;
        }
    }

    return found;
}

// ====================测试代码====================
void Test(char* testName, int* matrix, int rows, int columns, int number, bool expected)
{
    if(testName != NULL)
        printf("%s begins: ", testName);

    bool result = Find(matrix, rows, columns, number);
    if(result == expected)
        printf("Passed.\n");
    else
        printf("Failed.\n");
}

//  1   2   8   9
//  2   4   9   12
//  4   7   10  13
//  6   8   11  15
// 要查找的数在数组中
void Test1()
{
    int matrix[][] = {{, , , }, {, , , }, {, , , }, {, , , }};
    Test("Test1", (int*)matrix, , , , true);
}

//  1   2   8   9
//  2   4   9   12
//  4   7   10  13
//  6   8   11  15
// 要查找的数不在数组中
void Test2()
{
    int matrix[][] = {{, , , }, {, , , }, {, , , }, {, , , }};
    Test("Test2", (int*)matrix, , , , false);
}

//  1   2   8   9
//  2   4   9   12
//  4   7   10  13
//  6   8   11  15
// 要查找的数是数组中最小的数字
void Test3()
{
    int matrix[][] = {{, , , }, {, , , }, {, , , }, {, , , }};
    Test("Test3", (int*)matrix, , , , true);
}

//  1   2   8   9
//  2   4   9   12
//  4   7   10  13
//  6   8   11  15
// 要查找的数是数组中最大的数字
void Test4()
{
    int matrix[][] = {{, , , }, {, , , }, {, , , }, {, , , }};
    Test("Test4", (int*)matrix, , , , true);
}

//  1   2   8   9
//  2   4   9   12
//  4   7   10  13
//  6   8   11  15
// 要查找的数比数组中最小的数字还小
void Test5()
{
    int matrix[][] = {{, , , }, {, , , }, {, , , }, {, , , }};
    Test("Test5", (int*)matrix, , , , false);
}

//  1   2   8   9
//  2   4   9   12
//  4   7   10  13
//  6   8   11  15
// 要查找的数比数组中最大的数字还大
void Test6()
{
    int matrix[][] = {{, , , }, {, , , }, {, , , }, {, , , }};
    Test("Test6", (int*)matrix, , , , false);
}

// 鲁棒性测试，输入空指针
void Test7()
{
    Test("Test7", NULL, , , , false);
}

int _tmain(int argc, _TCHAR* argv[])
{
    Test1();
    Test2();
    Test3();
    Test4();
    Test5();
    Test6();
    Test7();

    return ;
}

5. 项目 03_FindInPartiallySortedMatrix 下载

百度网盘： 03_FindInPartiallySortedMatrix.zip

参考资料

[1]  何海涛. 剑指 Offer：名企面试官精讲典型编程题 [M]. 北京：电子工业出版社，2012. 38-41.