查找矩阵中是否存在一个数

来源:互联网 发布:传智播客java培训 编辑:程序博客网 时间:2024/04/29 04:19

题目意思:给定一个NxN的数字矩阵,每一行从左到右增加,每一列从上到下增加。查找其中是否存在一个数key。


算法思路:

  • 起始从右上角开始查找,a[i][j]初试值为a[0][n-1],循环下列
  • while( i < n && j >= 0)
    • 如果key < a[i][j],往左走,j–,
    • 如果key > a[i][j],则往下走,执行i++
    • 如果key == a[i][j],表示找到了

代码:

bool findNumber(vector <vector<int> > &a,int key){    int m = a.size();    bool flag = false;    int i = 0,j = m-1;    while(j >= 0 && i < m){        cout<<i<<"-"<<j<<endl;        if(key > a[i][j]){            i++;        } else if(key < a[i][j]){            j--;        } else{            flag = true;            break;        }    }    return flag;}
0 0
原创粉丝点击