题目描述
有一个NxM的整数矩阵,矩阵的行和列都是从小到大有序的。请设计一个高效的查找算法,查找矩阵中元素x的位置。
给定一个int有序矩阵mat,同时给定矩阵的大小n和m以及需要查找的元素x,请返回一个二元数组,代表该元素的行号和列号(均从零开始)。保证元素互异。
测试样例:
[[1,2,3],[4,5,6]],2,3,6
返回:[1,2]
这道题leetcode上也有类似的,有很多优化的解法,会对自己码代码帮助很多,但是过了体验期,就要30美元一个月。
总的来说还是比较简单的。
import java.util.*;
public class Finder {
public int[] findElement(int[][] mat, int n, int m, int x) {
// write code here
int row = 0;
int col =m-1;
while(row<n&&col>=0) {
if(x>mat[row][col])row++;
if(x==mat[row][col]) {
return new int[] {row,col};
}
if(x<mat[row][col])col--;
}
return null;
}
}