天气变化大,生病折腾了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;//row
int n;//col
int t;//needed number
int 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就不会超时。