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

题目意思：给定一个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;
}