博客
关于我
LeetCode 378.有序矩阵中第K小的元素
阅读量:246 次
发布时间:2019-03-01

本文共 1402 字,大约阅读时间需要 4 分钟。

给你一个 n x n 矩阵 matrix ,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。

请注意,它是 排序后 的第 k 小元素,而不是第 k 个 不同 的元素。

用最小堆维护候选集合

每次堆中取出一个元素 将它的右元素和下元素加入候选集合
用数组判断某个元素是否已经被加入过堆

class Solution {       class Node{           int x;        int y;        int val;        public Node(int x, int y, int val){               this.x = x;            this.y = y;            this.val = val;        }    }    class NodeComparator implements Comparator
{ public int compare(Node a, Node b){ return a.val - b.val; } } public int kthSmallest(int[][] matrix, int k) { int n = matrix.length; int[] cx = { 0, 1}; int[] cy = { 1, 0}; boolean[][] hash = new boolean[n][n]; Queue
minheap = new PriorityQueue<>(k, new NodeComparator()); minheap.add(new Node(0, 0, matrix[0][0])); int count = 0; while(!minheap.isEmpty()){ Node node = minheap.poll(); if(++count == k) return matrix[node.x][node.y]; for(int i = 0; i < 2; i++){ if(node.x + cx[i] < n && node.y + cy[i] < n && !hash[node.x + cx[i]][node.y + cy[i]]){ minheap.offer(new Node(node.x + cx[i], node.y + cy[i], matrix[node.x + cx[i]][node.y + cy[i]])); hash[node.x + cx[i]][node.y + cy[i]] = true; } } } return 0; }}

转载地址:http://rtvv.baihongyu.com/

你可能感兴趣的文章
Mysql学习总结(2)——Mysql超详细Window安装教程
查看>>
Mysql学习总结(30)——MySQL 索引详解大全
查看>>
Mysql学习总结(31)——MySql使用建议,尽量避免这些问题
查看>>
Mysql学习总结(33)——阿里云centos配置MySQL主从复制
查看>>
Mysql学习总结(35)——Mysql两千万数据优化及迁移
查看>>
Mysql学习总结(36)——Mysql查询优化
查看>>
Mysql学习总结(37)——Mysql Limit 分页查询优化
查看>>
Mysql学习总结(38)——21条MySql性能优化经验
查看>>
Mysql学习总结(39)——49条MySql语句优化技巧
查看>>
Mysql学习总结(3)——MySql语句大全:创建、授权、查询、修改等
查看>>
Mysql学习总结(40)——MySql之Select用法汇总
查看>>
Mysql学习总结(41)——MySql数据库基本语句再体会
查看>>
Mysql学习总结(42)——MySql常用脚本大全
查看>>
Mysql学习总结(43)——MySQL主从复制详细配置
查看>>
Mysql学习总结(44)——Linux下如何实现mysql数据库每天自动备份定时备份
查看>>
Mysql学习总结(45)——Mysql视图和事务
查看>>
Mysql学习总结(46)——8种常被忽视的SQL错误用法
查看>>
Mysql学习总结(48)——MySql的日志与备份还原
查看>>
Mysql学习总结(49)——从开发规范、选型、拆分到减压
查看>>
Mysql学习总结(4)——MySql基础知识、存储引擎与常用数据类型
查看>>