-
在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
本题来自《剑指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; }