《剑指Offer》——二维数组中的查找

题目描述

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

 

#include<cstdio>
#include<cstdlib>
#include<vector>
#include<iostream>
using namespace std;
/*
* Created by HarvestWu on 2019/01/14.
/*
矩阵有序，从左下角开始，向上递减，向右递增；
从左下角开始查找，当要查找数字比左下角数字大时，右移；
要查找数字比左下角数字小时，上移。
*/
class Solution {
public:
	bool Find(int target, vector<vector<int> > array) {
		int rowNum = array.size();
		int colNum = array[0].size();
		int i, j;
		for (i = rowNum - 1, j = 0; i >= 0 && j < colNum;){
			if (target<array[i][j])
				i--;
			else if (target>array[i][j])
				j++;
			else
				return true;
		}
		return false;
	}
};

int main(){
	vector<vector<int>> array(5, vector<int>(5));
	int a = 0, b;
	for (int i = 0; i < 5; i++){
		b = a;
		for (int j = 0; j < 5; j++){
			array[i][j] = b++;
		}
		a++;
	}
	Solution s;
	cout << s.Find(45, array) << endl;
	return 0;
}