今天在公司遇到一个问题,需要对容器存储的结构体中的元素进行去重,这里用到了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;
}