题目
在一个 n * m 的二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
1.最普通的解法:
class Solution { public: bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) { // 从左到右递增,从上到下递增 for(int i = 0; i < matrix.size(); i++) { int size_rol = matrix[i].size(); for(int j = 0; j < size_rol ; j++) { if(matrix[i][j] == target) { return true; } // if (j > 1 && i > 1) // { // if(matrix[i-1][j-1] < target && matrix[i-1][j] > target && matrix[i][j-1] > target) // { // return false; //} // } } } return false; } };
2.进阶解法:
class Solution { public: bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) { // 获取行数、列数 int row_size = matrix.size(); int col_size = matrix[0].size(); if(row_size == 0) { return false; } if (col_size == 0) { return false; } for(int i = 0; i < row_size && i< col_size; i++) { if(matrix[i][i] == target) { return true; } if(matrix[i][i] > target) { for(int j = 0; j < i; j++) { if(matrix[j][i] == target || matrix[i][j] == target) { return true; } } return false; } } if(row_size > col_size) { for(int t_ = col_size; t_ < row_size ; t_++) { for(int k_ = 0; k_ < row_size; k_++) { if(matrix[k_][t_] == target) { return true; } } } return false; } if(row_size < col_size) { for(int t_ = row_size; t_ < col_size ; t_++) { for(int k_ = 0; k_ <col_size; k_++) { if(matrix[t_][k_] == target) { return true; } } } return false; } return false; } };