关注
#include <iostream>
#include<string>
#include <list>
#include <vector>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>
#include <iomanip>
using namespace std;
// 插入排序
template<class randomaccessiterator, class T>
void _unguarded_insert_aux(randomaccessiterator last, T value) {
randomaccessiterator next = last - 1;
while (*next >value) {
*last = *next;
last = next;
--next;
}
*last = value;
}
template<class randomaccessiterator>
void _insert_aux(randomaccessiterator first, randomaccessiterator last) {
typedef typename std::iterator_traits<randomaccessiterator>::value_type T;
T value = *last;
if (*first > value) {
std::copy_backward(first, last, last + 1);
*first = value;
}
else _unguarded_insert_aux(last, value);
}
template<class randomaccessiterator>
void _insert_sort(randomaccessiterator first, randomaccessiterator last) {
randomaccessiterator tmp = first + 1;
while (tmp != last) {
_insert_aux(first, tmp);
++tmp;
}
}
//求 数组首、中、尾元素的中位数,防止分割恶化
template<class T>
inline const T& _median(const T&a, const T&b, const T&c) {
if (a < b) {
if (b < c) return b;
else if (a<c) return c;
else return a;
}
else {
if (c > a) return a;
else if (c>b)return c;
else return b;
}
}
//做分割处理
template<class randomaccessiterator, class T>
randomaccessiterator t_pivot_partition(randomaccessiterator first, randomaccessiterator last, T pivot) {
while (true) {
while (*first < pivot) ++first;
--last;
while (*last > pivot) --last;
if (!(first < last)) return first;
std::swap(*first, *last);
++first;
}
}
//求区间第k小的元素
template <class RandomAccessIterator>
void my_nth_element(RandomAccessIterator first, RandomAccessIterator nth, RandomAccessIterator last) {
while (last - first >= 3) {
RandomAccessIterator tmp = t_pivot_partition(first, last, _median(*first, *(last - 1), *(first + (last - first) / 2)));
if (tmp <= nth)
first = tmp;
else last = tmp;
}
_insert_sort(first, last);
}
//包装函数
int KthArray(vector<int>& nums,int k) {
my_nth_element(nums.begin(),nums.begin()+k,nums.end());
return *(nums.begin() + k);
}
int main() {
int n;
vector<int> vec;
while (cin>>n) {
vec.push_back(n);
}
int k = vec[vec.size() - 1];
vec.pop_back();
cout << KthArray(vec, vec.size()-k)<< endl;
return 0;
}
查看原帖
点赞 评论
相关推荐
03-11 18:10
北京信息科技大学 材料工程师 点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 笔试 #
2070527次浏览 24128人参与
# 读研or工作,哪个性价比更高? #
29902次浏览 386人参与
# 携程求职进展汇总 #
183015次浏览 1263人参与
# 牛友故事会 #
184257次浏览 3317人参与
# 讲讲我的真实离职原因 #
26797次浏览 181人参与
# 传音控股求职进展汇总 #
8546次浏览 61人参与
# 元戎现在香不香 #
66468次浏览 541人参与
# 歌尔求职进展汇总 #
48110次浏览 321人参与
# 90后北漂现状 #
20430次浏览 186人参与
# 德州仪器求职进展汇总 #
1555次浏览 20人参与
# 安克创新求职进展汇总 #
20576次浏览 199人参与
# 初创公司值得加入吗? #
14986次浏览 112人参与
# 实习必须要去大厂吗? #
75392次浏览 1130人参与
# 软开人,秋招你打算投哪些公司呢 #
72330次浏览 798人参与
# 打杂的实习你会去吗? #
101990次浏览 909人参与
# 24届营销人拿到了几个offer #
11796次浏览 102人参与
# 技术岗笔试题求解 #
32306次浏览 504人参与
# 入职以后才知道的校招谎言 #
67299次浏览 415人参与
# OPPO求职进展汇总 #
602234次浏览 4751人参与
# 牛友打假中心 #
5211次浏览 256人参与
# 滴滴求职进展汇总 #
116831次浏览 974人参与
# 如果不工作真的会快乐吗 #
89266次浏览 747人参与