二维数组中查找某个数是否存在

题目很明了,给一个二位数组,二维数组从左到右逐渐增大,从上到下逐渐增大,再给一个要查找的数,判断数组里是否存在该数字。

最无脑的版本

下面的解法应该是排除直接双重循环之后最无脑的解法了,每次都先对一维数组进行判断,如果array[i][0] <= target && array[i][array[0].length - 1] >= target,就对这个数组进行一次遍历,查找目标数,如果存在返回true,否则进行下一次循环。

public static boolean Find(int target, int[][] array) {
        if (array == null) {
            return false;
        }
        if (array.length <= 0) {
            return false;
        }

        for (int i = 0; i < array.length; i++) {
            if (array[i][0] <= target && array[i][array[0].length - 1] >= target) {
                for (int j = 0; j < array[0].length; j++) {
                    if (target == array[i][j]) {
                        return true;
                    }
                }
            }
        }
        return false;
    }

稍微动点脑子

上面的程序明显可以优化,最坏情况还是O(n^2),即每个一维数组都符合if中的条件,从而进行一次查找。那么查找程序是否还可以优化呢,利用题目的条件:从左到右是递增的,那么利用二分查找可以将程序的最坏复杂度优化至O(nlogn)

if (array == null) {
            return false;
        }
        if (array.length == 0) {
            return false;
        }
        if (array[0].length <= 0) {
            return false;
        }

        for (int i = 0; array.length > 0 && i < array.length; i++) {
            if (array[i][0] <= target && array[i][array[0].length - 1] >= target) {
                int x = 0, y = array[0].length - 1, mid = (x + y) / 2;
                while (x <= y) {
                    if (array[i][mid] > target) {
                        y = mid - 1;
                        mid = (x + y) / 2;
                    } else if (array[i][mid] < target) {
                        x = mid + 1;
                        mid = (x + y) / 2;
                    } else {
                        return true;
                    }
                }
            }
        }
        return false;

更简单的方法

在讨论区找到了更简单的方法,通过从左下角开始搜索

public static boolean Find(int target, int[][] array) {
        int len = array.length - 1;
        int i = 0;
        while ((len >= 0) && (i < array[0].length)) {
            if (array[len][i] > target) {
                len--;
            } else if (array[len][i] < target) {
                i++;
            } else {
                return true;
            }
        }
        return false;
}

这种算法下如果是一个M x N的二维数组,复杂度为O(M + N)。同样的思路从右下角开始搜索也是可以的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,788评论 0 33
  • LeetCode 刷题随手记 - 第一部分 前 256 题(非会员),仅算法题,的吐槽 https://leetc...
    蕾娜漢默阅读 17,962评论 2 36
  • 一维数组 首先开始最基本的Binary Search, 数组是有序的,但是有重复数。例题: Search for ...
    dol_re_mi阅读 2,424评论 0 2
  • 转眼已是明媚四月,天气渐渐暖和起来,春光正好,去赴一场花的盛会,心若向阳,又何惧悲伤。 离开烟台将近一个月的时间,...
    映嘉荷阅读 386评论 0 2
  • 很快,这个学期马上就结束了,恐怖的期末考试也来了。图书馆自习的人史无前例地多起来。 Zac说他不喜欢人太多,注意力...
    苏戈的假想敌阅读 270评论 0 0