ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

在行列都排好序的矩阵中找指定的数

2022-01-09 19:00:41  阅读:153  来源: 互联网

标签:matrix nextInt int 矩阵 排好序 行列 sc


题目

在行列都排好序的矩阵中找指定的数

描述

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

输入描述:

第一行有三个整数N, M, K
接下来N行,每行M个整数为输入的矩阵

输出描述:

若K存在于矩阵中输出"Yes",否则输出"No"

示例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

思路:

一般的思路是从右上角或者左下角开始查找,如果k大于右上角的值,就排除第一行,反之排除最后一列,如此反复进行,这样算法的时间复杂度就是O(M + N)。

import java.util.Scanner;


public class Main{
    
    public static boolean ifKInMatrix(int[][] matrix,int k) {
        int rows = matrix.length;
        int cols = matrix[0].length;
        
        int i = 0;
        int j = cols -1;
        while(i<rows && j>=0){
            if(matrix[i][j] < k) {
                i++;
            } else if(matrix[i][j] >k) {
                j--;
            } else {
                return true;
            }
        }
        return false;
    }
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int k = sc.nextInt();
        
        int[][] matrix = new int[n][m];
        for (int i=0;i < n;i++){
            for (int j=0;j < m;j++){
                matrix[i][j] = sc.nextInt();
            }
        }
        
        boolean res = ifKInMatrix(matrix,k);
        if(res){
            System.out.println("Yes");
        } else {
            System.out.println("No");
        }
    }
}

标签:matrix,nextInt,int,矩阵,排好序,行列,sc
来源: https://www.cnblogs.com/sfnz/p/15781855.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有