二维数组中的查找:最直观的想法是,两层for循环直接暴力查找,很显然超时,那么考虑到二维数组从左到右有序,从上到下有序,则使用深度优先搜索dfs,当array[i][j]小于target时,向右搜或者向下搜,很显然也超时,故使用记忆化搜索ms,用一个二维数组memo来记录dfs[i][j]的结果,通过。
bool dfs(vector<vector<int>> &memo,vector<vector<int>> &array,int target,int i,int j) { if(i>=array.size()||j>=array[0].size()) return false; if(memo[i][j]!=-1) //如果memo为bool类型反而不好初始化 return memo[i][j]; if(array[i][j]>target) return false; if(array[i][j]==target) return true; if(array[i][j]<target) memo[i][j] = dfs(memo,array,target,i+1,j)||dfs(memo,array,target,i,j+1); return memo[i][j]; //int与bool的隐式转换 } bool Find(int target, vector<vector<int> > array) { vector<vector<int>> memo(array.size(),vector<int>(array[0].size(),-1)); return dfs(memo,array,target,0,0); }
优化:如果是在有序的一维数组中查找,那么会想到二分查找,小于则在中点向左查找,大于则在中点向右查找,那么在有序的二维数组中又该如何查找呢?将中点设置在右上角,小于则在中点向左查找,大于则在中点向下查找。
bool Find(int target, vector<vector<int> > array) { int m=array.size(),n=array[0].size(); int i=0,j=n-1; while(i<m&&j>=0) { if(array[i][j]==target) return true; else if(array[i][j]>target) j--; else i++; } return false; }