有序矩阵查找 -- 算法小结

来源:互联网 发布:arduino matlab编程 编辑:程序博客网 时间:2024/05/19 01:13

现在有一个行和列都排好序的矩阵,请设计一个高效算法,快速查找矩阵中是否含有值x。

给定一个int矩阵mat,同时给定矩阵大小nxm及待查找的数x,请返回一个bool值,代表矩阵中是否存在x。所有矩阵中数字及x均为int范围内整数。保证n和m均小于等于1000。

测试样例:
[[1,2,3],[4,5,6],[7,8,9]],3,3,10
返回:false
编程思路:由于矩阵已经是有序的了,我们可以从右上角开始,当当前位置的数比目标数大时,可以知道,当前位置的数及所在列的下方位置的数都不符合,将列数往里缩减1列;比目标数小时,则当前位置的数及所在行的左方位置的数都不符合,将行数往下增一行。(其中,当找到与目标数相等的数时,返回TRUE,若至最后行列均到底,还未找到,怎返回FALSE)

import java.util.*;public class Finder {    public boolean findX(int[][] mat, int n, int m, int x) {        // write code here        if(m<1||n<1){            return false;        }        int i=n-1,j=0;        while(i>=0&&j<m){  //判断边界           if(mat[i][j]>x){                  i--;    //当前数大于目标数,列数-1           }else if(mat[i][j]<x){               j++;     //当前数小于目标数,行数+1           }else{               return true;   //找到           }        }        return false;   //未找到    }}

这里写图片描述