【洛谷 P1923】【深基9.例4】求第 k 小的数题解（nth_element）

最新推荐文章于 2024-07-02 09:05:18 发布

HEX9CF

最新推荐文章于 2024-07-02 09:05:18 发布

阅读量179

点赞数

分类专栏： Algorithm Problems 文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/qq_34988204/article/details/129281415

版权

Algorithm Problems 专栏收录该内容

462 篇文章 13 订阅

订阅专栏

该文章是一个关于编程算法的问题，涉及寻找一个奇数数量（n）的数字序列中第k小的数字。通过使用C++中的nth_element函数，可以有效地找到这个值。给定的示例程序读取输入，存储数字在向量中，然后使用nth_element找出并输出第k小的数字。

摘要由CSDN通过智能技术生成

【深基9.例4】求第 k 小的数

题目描述

输入 $n$ （ $\le n < 5000000$ 且 $n$ 为奇数）个数字 $a_i$ （ $\le a_i < {10}^9$ ），输出这些数字的第 $k$ 小的数。最小的数是第 $0$ 小。

输入格式

输出格式

样例 #1

样例输入 #1

5 1
4 3 2 1 5

样例输出 #1

思路

用nth_element求第 k 小的数。

AC代码

#include <iostream>
#include <vector>
#include <algorithm>
#define AUTHOR "HEX9CF"
using namespace std;

vector<int> v;

void read(int &x)
{
    char ch;
    x = 0;
    while (('0' > ch || '9' < ch))
    {
        ch = getchar();
    }
    while (!('0' > ch || '9' < ch))
    {
        x = x * 10 + ch - '0';
        ch = getchar();
    }
}

int main()
{
    int n, k;
    read(n);
    read(k);
    for (int i = 0; i < n; i++)
    {
        int in;
        read(in);
        v.push_back(in);
    }
    nth_element(v.begin(), v.begin() + k, v.end());
    cout << v[k] << endl;
    return 0;
}