二维数组查找

来源:互联网 发布:java优秀开源项目 编辑:程序博客网 时间:2024/05/21 13:48

天气变化大,生病折腾了1个礼拜,写篇博客压压惊。

题目源自于清明假期前夕舍友春招的笔试题,后来得知是剑指offer上的一道题,书上应该有解题思路啥的。

题目重述

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


由于从左到右,从上到下是递增的。如果从索引[0,0]开始,比当前索引[i,j]对应的数值大的数可能在下一行[i+1,j]也可能在下一列[i,j+1]。所以,查找起始索引设定在[0,col-1]。

这样讲所需要查找的key与 当前索引[i,j]对应的值比较时:当 key>a[i,j]。则key一定在下一行 key < a[i,j],则key一定在上一列。类似于二叉排序树。


代码

#include <iostream>using namespace std;int a[1001][1001];int m;//rowint n;//colint t;//needed numberint main(){int i,j;cin>>m>>n>>t;for(i=0;i<m;i++){for(j=0;j<n;j++){cin>>a[i][j];}}i = 0;j = n - 1;while (1){if(t > a[i][j]){i++;if(i > m-1){cout << "No" <<endl;break;}}if(t < a[i][j]){j--;if(j<0){cout << "No" <<endl;break;}}if(t == a[i][j]){cout << "Yes" <<endl;break;}}return 0;}

注:上述算法据说在九度OJ上用C++会超时,改成C就不会超时。


0 0
原创粉丝点击