HDU-6040 Hints of sd0061（线性找第k小）

最新推荐文章于 2017-09-27 04:16:00 发布

丿残念灬

最新推荐文章于 2017-09-27 04:16:00 发布

阅读量809

点赞数

CC 4.0 BY-SA版权

分类专栏：暴力文章标签：线性找第k小

本文链接：https://blog.youkuaiyun.com/qq_31759205/article/details/76154626

暴力专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种使用C++ STL中的nth_element函数来高效查找数组中第K小元素的方法，并通过一道具体题目（HDU-6040）进行实现与解析，展示了如何利用该函数优化查询过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门：HDU-6040

题意：给出n个数字，由m次查询，第i次查询问第b[i]小的数是多少

题解：线性找第k小

用到一个STL

nth_element(A, A + k, A+n)

表示在数组A的[0,n-1]中找到第k小的并放在第k个位置，并且前k-1个位置均为小于A[k]的数（要注意的一点就是：第k小中k是从0开始算的）

注意到在找到第k大的时候，前k-1个数均是小于A[k]的，这样我们从大往小枚举，可以减少搜索量

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MX = 1e7 + 5;

struct node {
    int k, id;
    bool operator<(const node& _A)const {
        return k < _A.k;
    }
} B[105];
unsigned A[MX];
int n, m, cas = 0;
unsigned x, y, z, a, b, c, ans[105];

unsigned cal() {
    unsigned t;
    x ^= x << 16; x ^= x >> 5; x ^= x << 1;
    t = x; x = y; y = z;
    z = t ^ x ^ y;
    return z;
}
int main() {
    //freopen("in.txt","r",stdin);
    while (~scanf("%d%d%u%u%u", &n, &m, &a, &b, &c)) {
        x = a; y = b; z = c;
        for (int i = 0; i < m; i++) scanf("%d", &B[i].k), B[i].id = i;
        for (int i = 0; i < n; i++) A[i] = cal();
        sort(B, B + m);
        B[m].id = m;
        B[m].k = n;
        for (int i = m - 1; i >= 0; i--) {
            nth_element(A, A + B[i].k, A + B[i + 1].k);
            ans[B[i].id] = A[B[i].k];
        }
        printf("Case #%d:", ++cas);
        for (int i = 0; i < m; i++) printf(" %u", ans[i]);
        printf("\n");
    }
    return 0;
}