使用STL去除std::vector自定義結構體重複項
阿新 • • 發佈:2019-02-08
最近搞了一個小東西要去除一個vector中重複的項。是這樣的:我用組播搜尋裝置,得到裝置IP資訊、版本號資訊,等,但有時會接收到多個相同IP裝置的資訊,因此要過濾掉重複的IP。我使用vector儲存每臺裝置資訊,包括IP、版本號,因此需要使用結構體。另外,要對這些裝置IP進行排序,讓其IP順序排列。
抽象出來,就是對一個包括結構體的vector進行排序、過濾重複項。參考網路資源,做了實現。
原始碼如下:
#include <vector> #include <string> #include <algorithm> #include <stdio.h> class foobar { public: std::string token; std::string item; int number; // 過載操作符 bool operator<(const foobar& rhs) { return (*this).token < rhs.token;}; bool operator>(const foobar& rhs) { return (*this).token > rhs.token;}; bool operator==(const foobar& rhs) { return (*this).token == rhs.token;} }; bool sort_token(const foobar& s1, const foobar& s2) { return s1.token.compare(s2.token) < 0; } int main(int argc, char* argv[]) { std::vector<foobar> vFoo; foobar tmp; tmp.token = "osd_1"; tmp.item = "OSD11111"; tmp.number = 1; vFoo.push_back(tmp); tmp.token = "osd_0"; tmp.item = "OSD111111"; tmp.number = 0; vFoo.push_back(tmp); tmp.token = "osd_2"; tmp.item = "OSD22222"; tmp.number = 2; vFoo.push_back(tmp); tmp.token = "osd_3"; tmp.item = "OS3330"; tmp.number = 3; vFoo.push_back(tmp); tmp.token = "osd_4"; tmp.item = "OS444444444444444"; tmp.number = 4; vFoo.push_back(tmp); tmp.token = "osd_1"; tmp.item = "OS5555555554444"; tmp.number = 5; vFoo.push_back(tmp); printf("before sort: \n"); for (unsigned int i = 0; i < vFoo.size(); i++) { printf("token: %s font: %d item: %s\n", vFoo[i].token.c_str(), vFoo[i].number, vFoo[i].item.c_str()); } std::sort(vFoo.begin(),vFoo.end(),sort_token); // 先排序 // 去除重複項 std::vector<foobar>::iterator unque_it = std::unique(vFoo.begin(), vFoo.end()); vFoo.erase(unque_it, vFoo.end()); printf("after unque: \n"); for (unsigned int i = 0; i < vFoo.size(); i++) { printf("token: %s font: %d item: %s\n", vFoo[i].token.c_str(), vFoo[i].number, vFoo[i].item.c_str()); } return 0; }
李遲 2016.6.9 週四端午 夜