stl vector自定义类型的去重问题

最近项目遇到一个问题，有关stl vector自定义类型的去重问题。

背景：1、在一个vector中，存在大量元素拥有同一属性，而其他属性我们不关心，为了减少数据包大小，需要去重

　　　2、此自定义类型不能去重载==操作符(公司代码规范等原因)

　　　3、正常情况下，vector中元素是有序的(拥有同一属性的元素排在一起)                /*引起误解，之后补充*/

于是，花了十分钟撸出了下列代码原型。

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

struct test
{
        int key;
        int value;
};

class finder
{
        public:
                explicit finder(const test& _t):t(_t){}
                const bool operator()(const test& __t)const{return t.key==__t.key;}
        private:
                test t;
};

int main(void)
{
        vector<test> vTest;
        test t1 = {1,10};
        test t2 = {1,11};
        test t3 = {2,8};
        test t4 = {2,9};
        test t5 = {1,6};
        vTest.push_back(t1);
        vTest.push_back(t2);
        vTest.push_back(t3);
        vTest.push_back(t4);
        vTest.push_back(t5);

        for(int index=0; index!=vTest.size(); ++index)
        {
                vTest.erase(find_if(vTest.begin(),vTest.end(),finder(vTest[index])));
                if(index == vTest.size())
                        break;
         }

        for(int index=0; index!=vTest.size(); ++index)
                cout<<vTest[index].key<<'\t'<<vTest[index].value<<endl;

        return 0;
}

结果：

这里利用了一下仿函数(functor)来构造find_if的条件，达到查找具有相同属性的自定义类型对象。

if里的判断是为了防止segment fault，erase在删除最后一个元素的时候，会给你带来惊喜-_-(详情请查看stl吧)。

目的达到了，但是总感觉循环太浪费CPU太低效太罪恶了，各位看官有没有更好的办法呢？求解。

 

================================================================

有小伙伴说用unique，好吧，其实之前也想到了，貌似效率没啥提升，代码还增加了。

贴代码(所有)：

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

struct test
{
    int key;
    int value;
};

class finder
{
  public:
    explicit finder(const test& _t):t(_t){}
    const bool operator()(const test& __t)const{return t.key==__t.key;}
  private:
    test t;
};
const bool unique_finder(const test& first, const test& second) 
{
    return first.key == second.key;
}
const bool sort_finder(const test& first, const test& second) 
{
    return first.key < second.key;
}

int main(void)
{
    vector<test> vTest;
    test t1 = {1,10}; 
    test t2 = {1,11}; 
    test t3 = {2,8};
    test t4 = {2,9};
    test t5 = {1,6};
    vTest.push_back(t1);
    vTest.push_back(t2);
    vTest.push_back(t3);
    vTest.push_back(t4);
    vTest.push_back(t5);

    /*
    for(int index=0; index!=vTest.size(); ++index)
    {
        vTest.erase(find_if(vTest.begin(),vTest.end(),finder(vTest[index])));
        if(index == vTest.size())
            break;
    }
    */

    cout<<"=================origin===================="<<endl;
    for(int index=0; index!=vTest.size(); ++index)
        cout<<vTest[index].key<<'\t'<<vTest[index].value<<endl;

    cout<<"=================sort===================="<<endl;
    sort(vTest.begin(),vTest.end(),sort_finder);
    for(int index=0; index!=vTest.size(); ++index)
        cout<<vTest[index].key<<'\t'<<vTest[index].value<<endl;

    /*
    cout<<"=================stable_sort===================="<<endl;
    stable_sort(vTest.begin(),vTest.end(),sort_finder);
    for(int index=0; index!=vTest.size(); ++index)
        cout<<vTest[index].key<<'\t'<<vTest[index].value<<endl;
    */

    cout<<"==================unique==================="<<endl;
    vTest.erase(unique(vTest.begin(),vTest.end(),unique_finder),vTest.end());

    for(int index=0; index!=vTest.size(); ++index)
        cout<<vTest[index].key<<'\t'<<vTest[index].value<<endl;

    return 0;
}

结果：

嘿嘿，不错，如果大家还有更好的办法，可以提出来，谢谢！

转载请注明出处：http://www.cnblogs.com/yuanzz/p/3714585.html