程序员代码指南-1-在行列都有序的矩阵中找指定数

1.题目描述

给定一个N×MN \times MN×M的整形矩阵matrix和一个整数K, matrix的每一行和每一列都是排好序的。 实现一个函数，判断K是否在matrix中 [要求] 时间复杂度为O(N+M)O(N+M)O(N+M)，额外空间复杂度为O(1)O(1)O(1)。

2. 输入描述:

第一行有三个整数N, M, K
接下来N行，每行M个整数为输入的矩阵
输出描述:
若K存在于矩阵中输出"Yes"，否则输出"No"

3.测试用例

示例1
输入
2 4 5
1 2 3 4
2 4 5 6
输出
Yes
示例2
输入
2 4 233
1 2 3 4
2 4 5 6
输出
No
备注:
1⩽N,M⩽10001 \leqslant N, M \leqslant 10001⩽N,M⩽10000⩽K,矩阵中的数⩽1090 \leqslant K, \text{矩阵中的数} \leqslant 10^90⩽K,矩阵中的数⩽109

4.思路分析

从右上角开始如果maxtrix[i][j] <k 排除这一行。因为右边的都小于当前数值，j++。边界为maxtrix[n-1][m-1]， maxtrix[i][j]>k 排除这一列，下面的都大于当前数值。i–.,边界为mtrix[0][0]。

5.Java代码

import  java.util.*;
public   class Main{
   public static void  main(String[] args){
       Scanner  scanner=new Scanner(System.in);
       int N =scanner.nextInt();
       int M=scanner.nextInt();
       int K=scanner.nextInt();
       int[][]maxtrix= new int[N][M];
       for(int i=0;i<N;i++){
           for(int j=0;j<M;j++){
               maxtrix[i][j]=scanner.nextInt();
           }
       }
       solution(maxtrix,K);
   } 
   public static  void  solution(int[][] arr,int k){
     if(arr==null){
        return ;
     } 
     //  System.out.println(arr.length+" "+arr[0].length);
       //起始位置
     int i=0;
     int j=arr[0].length-1;  
     while(true){
        // System.out.println("i="+i+" j= "+j);
       if(arr[i][j]==k){
           System.out.println("Yes");
           return;
       } else if(arr[i][j]<k ){//排除这-行
           if(i<arr.length-1)//行
           {
               i++;
           }else{
               break ;
           }   
       }else{
           if(j>0){
             j--;  
           }else{
               break ;
           } 
       }   
     }  
     
     System.out.println("No");  
   }  

6 测试结果