剑指offer--面试题3：二维数组的查找--Java实现

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

解题思路：
一般的查找方法，在OJ上都会超时，不管是顺序遍历还是每行二分查找等。
在这里，要结合这个二维数组的性质。行列都是有序的，所以可以选定左上角或者右下角开始比较。以左下角为例，如果待查元素较小，就向上查找，如果待查元素较大，就向右查找。

package com.java.offer;

import java.util.Scanner;

public class FindElementInMatrix {

    public static boolean find(int[] data, int rows, int columns, int val) {

        if (data != null && rows > 0 && columns > 0) {
            // 将坐标定位的右上角，也可以选为左下角
            int row = 0;
            int column = columns - 1;
            while (row < rows && column >= 0) {
                int tmp = data[row * columns + column];
                if (tmp == val) {
                    return true;
                } else if (tmp > val) {
                    column--;
                } else {
                    row++;
                }
            }
        }

        return false;
    }

    public static void main(String[] args) {

        int m = 0;// 行数
        int n = 0;// 列数

        int value = 0; // 待查找的值
        int[] data;

        Scanner in = new Scanner(System.in);


        while (in.hasNext()) {
            m = in.nextInt();// 行数
            n = in.nextInt();// 列数

            value = in.nextInt(); // 待查找的值
            data = new int[m * n];


            for (int i = 0; i < m * n; i++) {
                data[i] = in.nextInt();
            }
            if (find(data, m, n, value)) {
                System.out.println("Yes");
            } else {
                System.out.println("No");
            }
            data = null;
        }

    }

}

这个题思路很简单，但是这段代码在自己测试时没有发现问题，在九度OJ上一直是超时。相同的C++实现就不会超时。原因应该是输入引起的。如果有人了解，请告知。