c++:对结构体容器中,重复的元素去重问题

今天在公司遇到一个问题,需要对容器存储的结构体中的元素进行去重,这里用到了sort与unique,头文件需要加上algorithm

#include <iostream>
#include <vector>
#include<algorithm>
using namespace std;
struct stu
{
  int id;
  bool operator<(const stu &s1)//重载<操作符
  {
    return (*this).id < s1.id;
  }
  bool operator==(const stu &s1)//重载==操作符
  {
    return (*this).id == s1.id;
  }
  bool operator>(const stu &s1)//重载>操作符
  {
    return (*this).id > s1.id;
  }
};
struct stuSort
{
  bool operator()(const stu &s1,const stu &s2)
  {
    return s1.id < s2.id;
  }
};
int main()
{
  vector<struct stu> s;
  struct stu p;
  for(int j = 0; j < 3; j++)
  {
    for(int i = 0; i < 5;i++)
    {
      p.id = i;
      s.push_back(p);
      cout << s[i].id;
    }
  }
  cout << endl;
  struct stuSort sortFunc;
  std::sort(s.begin(),s.end(),sortFunc);
  s.erase(unique(s.begin(),s.end()),s.end());
  for(int i = 0; i<s.size();i++)
    cout << s[i].id << endl;
  return 0;
}

猜你喜欢

转载自www.cnblogs.com/cansun/p/9340377.html