c++中STL排序算法程序

最新推荐文章于 2024-03-22 20:23:34 发布

Augusdi

最新推荐文章于 2024-03-22 20:23:34 发布

阅读量2.1k

点赞数

分类专栏： STL标准模板库

本文链接：https://blog.csdn.net/Augusdi/article/details/8865508

版权

STL标准模板库专栏收录该内容

45 篇文章 1 订阅

订阅专栏

本文详细介绍了C++中nth_element函数的使用方法，通过该函数可以在不完全排序的情况下找到容器中指定排名的元素，适用于大规模数据处理场景。文中还通过实例演示了如何利用该函数解决大规模学生分数排名问题，显著提高了效率。

摘要由CSDN通过智能技术生成

// paixu.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include "iostream"
#include "vector"
#include <algorithm>
using namespace std;

struct student
{
    string name;
    int score;
};
bool compare(int a,int b)
{
    return a<b;
}



int _tmain(int argc, _TCHAR* argv[])
{   
	std::vector <int> V;
    for(int i=0;i<10;i++)
    {
        V.push_back(10-i);
    }
    nth_element(V.begin(),V.begin()+10,V.end(),compare);
	for(int i=0;i<10;i++){
		 cout<<V[i]<<endl;
	}

	return 0;
}

nth_element函数的作用是：求一个容器中第n（大/小）的元素。函数参数：nth_element(first,nth,last,compare);这个函数的参数有些怪异，它是求[first,last]这个区间中第n(大/小)的元素，这点必须要注意，否则很容易写出这种形式（我就想当然的写了）：nth_element(first,last,nth,compare);

这个函数看似没有什么用处，那么我们看一个例子：给你1000万个学生的信息（姓名，分数），输出分数排名在第k（总分高）学生的信息。
我们看到这个例题，首先会想到这个思路：先对所有学生的分数进行排序，然后输出student_book[k-1]（下标从0开始）。但是一共有1000万个学生，其实我们只想知道一个学生的排名，对整个学生手册排序实在是太浪费了。

那么后面我们会想到：使用partial_sort对学生手册进行部分排序，然后再输出student_book[k-1]，如果数据比较厉害，例如：1000万个学生，让你输出第1000万学生的信息。这就相当于让你全部排序，当然你也可以进行一些优化，将复杂度降到排序500万个数字。不过这其实还是太浪费了。此时，我们需要使用nth_element函数。
)
注意：nth_element函数会将第n（大/小）的元素放到第n的位置，且比第n（大/小）的元素会放到（右边/左边）。

我们看一下前一个例子的代码（1000万个学生也许太多了，对付100万个学生还是绰绰有余的）：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <string>
#include <vector>
#include <iostream>
using namespace std;
struct student
{
string name;
int score;
};
bool compare(student a,student b)
{
return a.score>b.score;
}
int main()
{
vector <student> V;
int n,k;
cin>>n;
for(int i=0;i<n;i++)
{
student t;
cin>>t.name>>t.score;
V.push_back(t);
}
cin>>k;
nth_element(V.begin(),V.begin()+k,V.end(),compare);
cout<<V[k-1].name<<" "<<V[k-1].score<<endl;
return 0;
}