在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
输入:
输入可能包含多个测试样例,对于每个测试案例,
输入的第一行为两个整数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
推荐指数:※
来源:http://ac.jobdu.com/problem.php?pid=1384
这也是海涛的剑指offer上的题目。把cin改成scanf吧。
第一次想到的解法:(其实在寻找行列上也可以使用二分查找)
#include<iostream>
#include<string.h>
#include<stdio.h>
#include<stdlib.h>
using namespace std;
int b_find(int *arr,int val,int length){
int low=0,high=length-1;
while(low<=high){
int mid=(low+high)/2;
if(arr[mid]==val)
return mid;
else if(arr[mid]>val)
high=mid-1;
else
low=mid+1;
}
return -1;
}
int main()
{
int m,n;
while(cin>>m>>n){
int val;
cin>>val;
int **arr,i,j;
arr=new int *[m];
for(i=0;i<m;i++){
arr[i]=new int [n];
for(j=0;j<n;j++)
scanf("%d",&arr[i][j]);
}
int flag=false;
int low=-1,high=-1;
for(i=0;i<m;i++)
if(arr[i][n-1]>=val){
low=i;
break;
}
for(i=m-1;i>=0;i--)
if(arr[i][0]<=val){
high=i;
break;
}
if(low!=-1&&high!=-1&&low<=high){
for(i=low;i<=high;i++){
if(b_find(arr[i],val,n)!=-1){
flag=true;
break;
}
}
}
if(flag==true)
printf("Yes\n");
else
printf("No\n");
}
return 0;
}
海涛的解法:
#include<iostream>
#include<string>
#include<stdio.h>
#include<stdlib.h>
using namespace std;
int main()
{
int m,n;
while(scanf("%d%d",&m,&n)!=EOF){
int **arr,i,j,val;
arr=new int*[m];
cin>>val;
for(i=0;i<m;i++){
arr[i]=new int [n];
for(j=0;j<n;j++)
scanf("%d",&arr[i][j]);
}
int p=0,q=n-1;
bool flag=false;
while(p<m&&q>=0){
if(arr[p][q]==val){
flag=true;
break;
}
else if(arr[p][q]<val)
p++;
else
q--;
}
if(flag==true)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
return 0;
}