牛客网--第K大元素

1.查找第k大的元素

题目描述

给定一个无序的整型数组A[n],数组大小大于等于3,允许有值相同的元素;请设计算法找到该数组排序后第三大的元素值并输出.

输入描述:

一个非空的整数数组(至少有3个元素,可正可负)

输出描述:

第三大的元素值

示例1

输入

复制

[1,2,3,4,5]

输出

复制

3

示例2

输入

复制

[1,1,2,2,3]

输出

复制

2

思路：将所有的元素放进优先队列，默认升序，然后出队列，若出队列的个数为k-1个，剩下的顶部元素就是要求的。

程序：

#include <iostream>
#include <queue>
using namespace std;
 int main(){
     priority_queue<int> p;
     while (getchar()!=']'){
         int t;
         cin >>t;
         p.push(t);
     }
     int y=0;
     while (y<2){
         p.pop();
         y++;
     }
     cout<<p.top();
     return 0;
 }

 

2.

题目描述

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第k小的元素。
请注意，它是排序后的第k小元素，而不是第k个元素。

示例:

matrix = [
   [ 1,  5,  9],
   [10, 11, 13],
   [12, 13, 15]
],
k = 8,

返回 13。
说明: 
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

输入描述:

第一行为k的值和矩阵的n的值

后续为n*n矩阵的数字，以空格分割

输出描述:

矩阵中第k小的元素

示例1

输入

复制

8 3
1 5 9
10 11 13
12 13 15

输出

复制

13

程序：

#include<iostream>
#include<queue>
using namespace std;
int main()
{
    int k,n;
    while(cin>>k>>n)
    {
        priority_queue<int> p;
        int x;
        while(cin>>x)
        {
            p.push(x);
            if(p.size()>k)
                p.pop();
        }
        cout<<p.top()<<endl;
    }
    return 0;
}