关注
#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;
}
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
点赞 评论 收藏
分享
09-26 17:53
广东技术师范大学 前端工程师 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 你的mentor是什么样的人? #
13470次浏览 98人参与
# 牛客周边新品开箱 #
9826次浏览 89人参与
# 快手技术岗信息交流阵地 #
507次浏览 6人参与
# 怎么给家人解释你的工作? #
9033次浏览 61人参与
# 牛友的志愿填报指南 #
33927次浏览 183人参与
# 帮我看看,领导说这话什么意思? #
17694次浏览 92人参与
# 求职中的尴尬瞬间 #
2418次浏览 32人参与
# 求职低谷期你是怎么度过的 #
11225次浏览 230人参与
# 26届秋招公司红黑榜 #
25563次浏览 104人参与
# 校招泡的最久的公司是哪家? #
11101次浏览 74人参与
# 从哪些方向判断这个offer值不值得去? #
14429次浏览 171人参与
# 牛客树洞,我想对你说 #
5996次浏览 80人参与
# 机械人集合!你是什么工程师? #
19419次浏览 91人参与
# 国企还是互联网,你怎么选? #
168489次浏览 1215人参与
# 得物app工作体验 #
27418次浏览 64人参与
# 你觉得mentor喜欢什么样的实习生 #
15109次浏览 402人参与
# 面试紧张时你会有什么表现? #
3329次浏览 37人参与
# 度小满求职进展汇总 #
11975次浏览 64人参与
# 小红书求职进展汇总 #
123781次浏览 966人参与
# 今年形式下双非本找得到工作吗 #
236400次浏览 1427人参与
# 没有家庭托举的我是怎么找工作的 #
17832次浏览 209人参与
查看6道真题和解析