面试题3：二维数组中的查找

[b]题目：[/b]

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

[b]分析：[/b]

1 2  8  9
2 4  9 12
4 7 10 13
6 8 11 15

以查找7为例，从右上角开始：
1.9大于7，下一次只需要在9的左边区域查找
2.8大于7，下一次只需要在8的左边区域查找
3.2小于7，下一次只需要在2的下边区域查找
4.4小于7，下一次只需要在4的下边区域查找

[b]代码：[/b]

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

package net.ecoolsoft.question;

public class SearchInArray {
	/**
	 * 在一个二维数组中，每一行都按照从左到右递增的顺序排序，
	 * 每一列都按照从上到下递增的顺序排序。请完成一个函数，
	 * 输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
	 * @param data 数组
	 * @param key 要查找的数字
	 * @return	找到返回true，未找到返回false
	 */
	public boolean findInArray(int[][] data, int key) {
		if(data == null || data.length == 0) {
			return false;
		}
		int i = 0;
		int j = data[0].length-1;
		while(i<data.length && j>=0) {
			if(data[i][j] == key) {
				return true;
			} else if(data[i][j] > key) {
				j--;
			} else {
				i++;
			}
		}
		return false;
	}
}

package net.ecoolsoft.question;

import junit.framework.Assert;

import org.junit.Test;

public class SearchInArrayTest {
	@Test
	public void findInArrayFailedTest() {
		int[][] data = {{1,2,8,9}, {2,4,9,12}, {4,7,10,13}, {6,8,11,15}};
		int key = 14;
		SearchInArray sArray = new SearchInArray();
		boolean result = sArray.findInArray(data, key);
		Assert.assertFalse(result);
	}

	@Test
	public void findInArrayTest() {
		int[][] data = {{1,2,8,9}, {2,4,9,12}, {4,7,10,13}, {6,8,11,15}};
		int key = 11;
		SearchInArray sArray = new SearchInArray();
		boolean result = sArray.findInArray(data, key);
		Assert.assertTrue(result);
	}
}

参考资料：《剑指Offer——名企面试官精讲典型编程题》