std::nth_elelment排序



本文转自：http://blog.csdn.net/readzw/article/details/8746108

[cpp] view plain copy print ?

1. template<class _RanIt, class _Pr> inline  
2.     void nth_element(_RanIt _First, _RanIt _Nth, _RanIt _Last, _Pr _Pred)  
3.   
4. template<class _RanIt> inline  
5.     void nth_element(_RanIt _First, _RanIt _Nth, _RanIt _Last)  

template<class _RanIt, class _Pr> inline
    void nth_element(_RanIt _First, _RanIt _Nth, _RanIt _Last, _Pr _Pred)

template<class _RanIt> inline
	void nth_element(_RanIt _First, _RanIt _Nth, _RanIt _Last)

该函数的作用为将迭代器指向的从_First 到 _last 之间的元素进行二分排序，以_Nth 为分界，前面都比 _Nth 小（大），后面都比之大（小）；但是两段内并不有序。

特别适用于找出前k个最大（最小）的元素。

例如：

[cpp] view plain copy print ?

1. vector<int> temp;  
2. //  
3. // calc the value of temp;  
4. //  
5.   
6. nth_element(temp.begin(), temp.begin()+10,temp.end());  
7.   
8. //   

vector<int> temp;
//
// calc the value of temp;
//

nth_element(temp.begin(), temp.begin()+10,temp.end());

// 

如今遇到一个问题：要对结构体按照自定义的方式进行排序，那么可以用到第一个方法了。

首先定义一个返回bool类型的比较函数，然后将函数名作为参数传入。

例如：现要求按照价格挑出最贵的10本书。

[cpp] view plain copy print ?

1. bool prizeCompare(const CBook &b1, const CBook &b2)  
2. {  
3.       return b1.prize>b2.prize;  
4. }  
5.   
6. // ......  
7.   
8. vector<CBook> books;  
9.   
10. //挑出最贵的10本书  
11.   
12. std::nth_element(books.begin(), books.begin()+10,books.end(), prizeCompare);  
13. books.resize(10);  
14.     

bool prizeCompare(const CBook &b1, const CBook &b2)
{
      return b1.prize>b2.prize;
}

// ......

vector<CBook> books;

//挑出最贵的10本书

std::nth_element(books.begin(), books.begin()+10,books.end(), prizeCompare);
books.resize(10);
  

更深一步，如今只对物理学的书进行排序，现有一个vector<int> physics 只记录了书在books中的序号。要求挑出10本最贵的物理学的书。

代码如下：

[cpp] view plain copy print ?

1. struct Cmp{  
2.     std::vector<Book> _list;  
3.     Cmp(const std::vector<Book> &_books):_list(_books){}  
4.     bool operator()(int idx1,int idx2)  
5.     {  
6.           return _list[idx1].prize>_list[idx2].prize;  
7.      }  
8.        
9.   }     
10. //......  
11.   
12. vector<CBook> books;  
13. vector<int> physics;  
14.   
15. for(int i=0;i<books.size();i++){ if(books[i].category == PHYSICS) physics.push_back(i);  
16. }  
17.   
18. //......  
19.   
20. std::nth_element(physics.begin(), physics.begin()+10,physics.end(), Cmp(books));  
21. physics.resize(10);  

struct Cmp{
    std::vector<Book> _list;
    Cmp(const std::vector<Book> &_books):_list(_books){}
    bool operator()(int idx1,int idx2)
    {
          return _list[idx1].prize>_list[idx2].prize;
     }
     
  }   
//......

vector<CBook> books;
vector<int> physics;

for(int i=0;i<books.size();i++){ if(books[i].category == PHYSICS) physics.push_back(i);
}

//......

std::nth_element(physics.begin(), physics.begin()+10,physics.end(), Cmp(books));
physics.resize(10);

[cpp] view plain copy print ?

1. <pre class="cpp" name="code"><pre></pre>  
2. <pre></pre>  
3. <pre></pre>  
4. <pre></pre>  
5. <pre></pre>  
6. <pre></pre>  
7. <pre></pre>  
8. <pre></pre>  
9. </pre>