leetCode No.378 Kth Smallest Element in a Sorted Matrix

最新推荐文章于 2016-11-16 12:42:36 发布

Andy_Shan

最新推荐文章于 2016-11-16 12:42:36 发布

阅读量355

点赞数

分类专栏： leetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/andy_shan/article/details/52443485

版权

leetCode 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

题目

Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.

Note that it is the kth smallest element in the sorted order, not the kth distinct element.

Example:

matrix = [ 

   [ 1,  5,  9], 

   [10, 11, 13], 

   [12, 13, 15] 

], 

k = 8,

return 13.

标签：Binary Search,Heap
相似题目： (M) Find K Pairs with Smallest Sums

题意

给定一个行和列都是从小到大排序的n维方阵。找到整个方阵中按从小到大排序第k个元素。

解题思路

假设当前点为（x，y）则当前点的相邻点为：（x，y+1），（x+1，y）。当前点的相邻点即为数值上比当前点大的相近的点。所以每次将相邻点放进优先队列（堆）直到达到想要得到的位置。

代码

import java.util.PriorityQueue;

public class Solution {
    public int kthSmallest(int[][] matrix, int k)
    {
        int m=matrix.length;
        if(m==0)
            return 0;

        int n=matrix[0].length;
        int cnt=0;
        boolean[][] used=new boolean[m][n];
        PriorityQueue<kthsNode> heap=new PriorityQueue<>();
        heap.add(new kthsNode(0, 0, matrix[0][0]));
        used[0][0]=true;
        kthsNode node=null;
        int[] xoff={0,1};
        int[] yoff={1,0};
        while(cnt<k)
        {
            node=heap.poll();
            int x=node.x;
            int y=node.y;
            for(int i=0;i<2;i++)
            {
                int xx=x+xoff[i];
                int yy=y+yoff[i];
                if(xx>=0&&xx<m&&yy>=0&&yy<n&&!used[xx][yy])
                {
                    used[xx][yy]=true;
                    heap.add(new kthsNode(xx, yy, matrix[xx][yy]));
                }
            }
            cnt++;
        }

        return node.val;
    }
}


class kthsNode implements Comparable<kthsNode>
{
    int x,y;
    int val;
    public kthsNode(int x,int y,int val)
    {
        this.x=x;
        this.y=y;
        this.val=val;
        // TODO Auto-generated constructor stub
    }
    @Override
    public int compareTo(kthsNode o)
    {
        // TODO Auto-generated method stub
        return val-o.val;
    }
}