解题报告：HDU_6040 Hints of sd0061 STL::nth_element

最新推荐文章于 2018-08-05 11:04:00 发布

原创最新推荐文章于 2018-08-05 11:04:00 发布 · 879 阅读

0 ·

CC 4.0 BY-SA版权

水题同时被 2 个专栏收录

23 篇文章

订阅专栏

数据结构

9 篇文章

订阅专栏

本文介绍了一种利用C++ STL中的nth_element函数高效查找序列中第k小元素的方法，并通过一个具体实例展示了如何在多次询问下，快速找到给定序列递推函数生成的序列上的第k小数。

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

题目链接

题意：

给你一个序列的递推函数，m次询问，询问这个序列上第bi小的数。

官方题解：

最关键的部分在于了解STL里的nth_element函数，简单介绍一下：

这个函数的时间复杂度近似线性

函数的调用(取第n小)：nth_element ( arr.begin() , arr+n，arr.end() ）

调用函数后保证比第n个数小的数都在前面，比它大的数在后面

代码：

#include<bits/stdc++.h>

using namespace std;

int id[105],A[105];
unsigned s[10000005];
unsigned ans[105];
int n,m;
unsigned x,y,z;

unsigned oper(){
   unsigned t ;
   x ^= x << 16;
   x ^= x >> 5;
   x ^= x << 1;
   t = x;
   x = y;
   y = z;
   z = t ^ x ^ y;
   return z;
}

bool cmp(int a,int b){
   return A[a] < A[b];
}

int main()
{
   int cas = 0;
   while(scanf("%d%d%u%u%u",&n,&m,&x,&y,&z)==5){
      for(int i=0;i<m;i++){
         id[i] = i;
         scanf("%d",&A[i]);
      }A[m]=n;id[m]=m;
      for(int i=0;i<n;i++)s[i] = oper();
      sort(id,id+m,cmp);
      for(int now,i=m-1,last=A[id[m]];i>=0;i--,last=now){
         now = A[id[i]];
         if(now==last){
            ans[id[i]] = ans[id[i+1]];
         }else {
            nth_element(s,s+now,s+last);
            ans[id[i]] = s[now];
         }
      }printf("Case #%d:",++cas);
      for(int i=0;i<m;i++){
         printf(" %u",ans[i]);
      }printf("\n");
   }return 0;
}