二维数组中的查找:最直观的想法是,两层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;
}



京公网安备 11010502036488号