当前位置：首页 > news >正文

[九度—剑指offer]—二维数组查找

news 来源：原创 2024/4/26 3:22:35

本题来自《剑指offer》面试题3

题目描述：

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

输入：

输入可能包含多个测试样例，对于每个测试案例，

输入的第一行为两个整数m和n(1<=m,n<=1000)：代表将要输入的矩阵的行数和列数。

输入的第二行包括一个整数t(1<=t<=1000000)：代表要查找的数字。

接下来的m行，每行有n个数，代表题目所给出的m行n列的矩阵(矩阵如题目描述所示，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。

输出：

对应每个测试案例，

输出”Yes”代表在二维数组中找到了数字t。

输出”No”代表在二维数组中没有找到数字t。

样例输入：

3 3
5
1 2 3
4 5 6
7 8 9
3 3
1
2 3 4
5 6 7
8 9 10
3 3
12
2 3 4
5 6 7
8 9 10

样例输出：

Yes
No
No

  思路：对于这种二维数组查找，如果从最左上角[0][0]向中间走，每一步都会有两种方向可能。这样问题越来越复杂。如果抱着从右上角或者左下角思路开始分析，显然每次可以排除一列的值。如果[i][j]比右上角的值小，则本列已经不可能有target了。如果大，则本行已经不可能有target了。这个问题启示我们当有多个路径可选时，换一个边界入手，形成相互约束，减少考虑分支。

 提交代码：Time Limit Exceed

<span style="font-size:12px;">#include<iostream>
using namespace std;
 
bool search(int (*a)[1000],int m,int n,int target)
{
    int i=0,j=n-1;
    while(i<m && n>0){
        if(a[i][j]==target)return 1;
        else {
            if(a[i][j]>target)j--;
            else
                i++;
            }
    }
    return 0;
}
 
 
int main()
{
    int m,n,target,a[1000][1000];
    while(cin>>m>>n)
    {
        cin>>target;
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                cin>>a[i][j];
        if(search(a,m,n,target))
            cout<<"Yes"<<endl;
        else
            cout<<"No"<<endl;
    }
    return 0;
}</span>

居然将cin改为scanf就不超时了

   
    #include<iostream>
#include <cstdio>
using namespace std;
 
bool search(int (*a)[1000],int m,int n,int target)
{
    int i=0,j=n-1;
    while(i<m && n>0){
        if(a[i][j]==target)return 1;
        else {
            if(a[i][j]>target)j--;
            else
                i++;
            }
    }
    return 0;
}
 
int main()
{
    int m,n,target,a[1000][1000];
    while(scanf("%d %d",&m,&n)!=EOF){
        scanf("%d",&target);
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&a[i][j]);
          if(search(a,m,n,target))
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}