STL之nth_element()(取容器中的第n大值)(不用在vector里也可以)
nth_element作用为求第m大的元素,通过近似于O(n)的时间复杂度将数组进行伪排序,保证第m个是数组中第m大的数,前面的都比第m个数小,后面的都比它大,但是无序的。
用法:
int a[10];
nth_element(a,a+5,a+9);
输出的是a[4],但是是第五大的数,从0开始,也可以改成从1开始,和sort类似【sort(a+1,a+n)】;
例:
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
int a[]={1,3,4,5,2,6,8,7,9};
int i;
cout<<"数列如下:"<<endl;
for(i=0;i<9;i++)
cout<<a[i]<<" ";
nth_element(a,a+5,a+9);
cout<<endl<<"输出第五大的数: "<<a[4]<<endl; //注意下标是从0开始计数的
return 0;
}
J Heritage of skywalkert
题目链接:
https://www.nowcoder.com/acm/contest/144/J
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn =1e7+20;
unsigned x,y,z,A,B,C;
typedef unsigned long long ull;
unsigned a[maxn];
unsigned tang(){
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()
{
int t,n;
cin>>t;
for(int cas=1;cas<=t;cas++){
cin>>n>>A>>B>>C;
x=A,y=B,z=C;
for(int i=1;i<=n;i++)
a[i]=tang();
vector<unsigned> vec;
int num=min(n,40);
nth_element(a+1,a+1+n-num,a+1+n);
for(int k=1;k<=num;k++){
vec.push_back(a[n+1-k]);
}
ull ans=0;
for(int i=0;i<vec.size();i++){
for(int j=i+1;j<vec.size();j++){
ull gcd = __gcd((ull)vec[i],(ull)vec[j]);
ull tp=vec[i]/gcd*vec[j];
ans = max(ans,tp);
}
}
printf("Case #%d: ",cas);
cout<<ans<<endl;
}
return 0;
}