二维数组中的查找

来源:互联网 发布:coc女王升级数据 编辑:程序博客网 时间:2024/04/27 21:48

《编程之美》上的一个问题:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。

public:
    bool Find(vector<vector<int> > array,int target) {
        //.......
    }

要解决这个问题,我相信很多人的第一想法是可以用递归,首先将target和array(m*n)左下角的元素array[m][0]比较:如果target<array[m][0],就可以排除最后一行,在剩下的array1(m-1*n)中寻找如果target>array[m][0],就可以排除第一列,在剩下的array1(m*n-1)中寻找。同理可以递归下去......

想法很可行,但是去掉一行或去掉一列这样的操作对vector<vector<int> > array的操作太复杂,实现起来不太容易。

再思考得到这样的算法:

bool Find(vector<vector<int> > array,int target) {
        int x = array.size();
        if(x == 0)
            return false;
        x--;
        int y = 0;
        while(x >= 0 && y < array[0].size())
        {
            if( array[x][y] == target )
                return true;
            else if( array[x][y] < target )
                y++;
            else              //array[x][y]>target
                x--;
        }
        return false;
    }

基本思想其实很类似于递归,但是是用x,y来标记位置,从左下角向右上角搜索,每次循环向上移动1,或者向右移动1,类似就排除一行或一列。最终会找到,或者到达终点右上角结束没找到。




1 0
原创粉丝点击