C++ std::map如何插入自定義的KEY


初學C++的小伙伴會問如果std::map中要使用自定義的key怎么辦?
答案重載描述符 "<",重載時請注意,當元素相等的時候要返回false.否則,插入相同的元素后,會生成多條記錄。而且使用find函數找不到自己的之前插入的key。

#include <stdio.h>
#include <map>
#include <iostream>
#include <string>

using namespace std;

struct A {
    int a;
    int b;
    int c;
    A(const int& a,const int& b,const int& c) {
       this->a = a; 
       this->b = b; 
       this->c = c; 
    }
    friend bool operator <(const A& l, const A& r);
};
void print (A tmp) {
    cout << "<" << tmp.a << "," <<  tmp.b << "," << tmp.c << ">";
}

bool operator <(const A& l, const A& r)
{
   if (l.a != r.a) { return l.a < r.a;}
   if (l.b != r.b) { return l.b < r.b;}
   if (l.c != r.c) { return l.c < r.c;}
   return false;
}

int main()
{
    A m(1,2,3);
    A n(1,2,3);
    A o(1,2,3);
    map<A,int> ss;    
    ss.insert(std::pair<A,int>(m,1));
    ss.insert(std::pair<A,int>(n,2));
    ss.insert(std::pair<A,int>(o,3));
    map<A,int>::iterator iter  = ss.begin();
    while (iter != ss.end())
    {
        cout << "KEY:";
        print(iter->first); 
        cout << ",VALUE:" << iter->second << endl; 
        iter ++;
    }
    iter = ss.find(m);
    if (iter != ss.end()) {
        cout << "Has find key." << endl;
        cout << "KEY:";
        print(iter->first); 
        cout << ",VALUE:" << iter->second << endl; 
    }else{
        cout << "Can't find key." << endl;
    }
    map<int,int> kk;
    kk.insert(std::pair<int,int>(1,1));
    kk.insert(std::pair<int,int>(1,2));
    kk.insert(std::pair<int,int>(1,3));
    map<int,int>::iterator iter2  = kk.begin();
    while (iter2 != kk.end())
    {
        cout << "KEY:" << iter2->first << ",VALUE:" << iter2->second << endl; 
        iter2 ++;
    }
    
    return 0;
}


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM