剑指offer编程题--二维数组中的查找(C++、Java解法)

题目描述

在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
时间限制：C/C++ 1秒，其他语言2秒 空间限制：C/C++ 32M，其他语言64M

c++解法

思路

1、我先设初始值val 为右上角元素
2、接下来进行二分操作：
3、如果val == target,直接返回
4、如果 target > val, 说明target在更大的位置，val左边的元素显然都是 < val，间接 < target ，说明第 0 行都是无效的，所以val下移一行。
5、如果 target< val, 说明target在更小的位置，val下边的元素显然都是 > val，间接 > tar，说明第 5 列都是无效的，所以val左移一列
6、继续步骤2

代码

class Solution {
public:
 bool Find(int target, vector<vector<int> > array) {
     // 判断数组是否为空
     int m = array.size();
     if (m == 0) return false;
     int n = array[0].size();
     if (n == 0) return false;
     int r = 0, c = n-1; // 右上角元素
     while (r<m && c>=0) {
         if (target == array[r][c]) {
             return true;
         }
         else if (target > array[r][c]) {
             ++r;
         }
         else {
             --c;
         }
     }
     return false;
 }
};

复杂度分析

时间复杂度：O(m+n) ，其中m为行数，n为列数，最坏情况下，需要遍历m+n次。
空间复杂度：O(1)

Java解法

思路

利用该二维数组的性质：
每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。
改变个说法，即对于左下角的值 m，m 是该行最小的数，是该列最大的数
每次将 m 和目标值 target 比较：
1.当 m < target，由于 m 已经是该列最大的元素，想要更大只有从列考虑，列的取值+1
2.当 m > target，由于 m 已经是该行最小的元素，想要更小只有从行考虑，行的取值-1
3.当 m = target，找到该值，返回 true
用某行最小或某列最大与 target 比较，每次可剔除一整行或一整列

代码

public class Solution {
    public boolean Find(int target, int [][] array) {
        int rows = array.length;
        if(rows == 0){
            return false;
        }
        int cols = array[0].length;
        if(cols == 0){
            return false;
        }
        // 左下
        int row = rows-1;
        int col = 0;
        while(row>=0 && col<cols){
            if(array[row][col] < target){
                col++;
            }else if(array[row][col] > target){
                row--;
            }else{
                return true;
            }
        }
        return false;
    }
}

复杂度分析

时间复杂度：O(m+n) ，其中m为行数，n为列数，最坏情况下，需要遍历m+n次。
空间复杂度：O(1)