C++ unordered_map/map的key

xytywh

已于 2022-05-10 19:17:51 修改

阅读量2.6k

点赞数 1

分类专栏： C/C++ 文章标签： c++

于 2021-09-13 19:45:11 首次发布

本文链接：https://blog.csdn.net/xiaohuihui1994/article/details/120273905

版权

C/C++ 专栏收录该内容

19 篇文章 1 订阅

订阅专栏

本文通过LeetCode的2001题为例，探讨了在C++中`unordered_map`和`map`的区别。`unordered_map`不支持`pair`或`vector`作为key，而`map`则可以。为使`unordered_map`能使用`pair`作为key，需要自定义哈希函数。解法展示了如何在不同场景下选择合适的数据结构来存储和处理数据。

摘要由CSDN通过智能技术生成

unordered_map可以使用基本类型(int等)，string作为key，不能使用pair vector作为key，map可以使用基本类型，string和pair vector作为key。

以leetcode-2001为例

// 解法1
class Solution {
public:
    int gcd(int a, int b) {
        if (a < b) return gcd(b, a);
        if (a % b == 0) return b;
        return gcd(b, a%b);
    }
    
    // 使用这种嵌套的方式表示两个key
    unordered_map<int, unordered_map<int, long long>> cnt;
    
    long long interchangeableRectangles(vector<vector<int>>& rectangles) {
        for (auto r : rectangles) {
            int c = gcd(r[0], r[1]);
            r[0] /= c;
            r[1] /= c;
            
            cnt[r[0]][r[1]]++;
        }
        
        long long ans = 0;
        
        for (auto iter = cnt.begin(); iter != cnt.end(); iter++) {
            for (auto i = iter->second.begin(); i != iter->second.end(); i++) {
                ans += i->second * (i->second - 1) / 2;
            }
        }
        
        
        return ans;
    }
};
// 解法2
class Solution {
public:
    int gcd(int a,int b)
    {
        if(b == 0)
            return a;
        return gcd(b,a%b);
    }
    long long interchangeableRectangles(vector<vector<int>>& rectangles) 
    {
    	// map可以直接使用pair作为key
        map<pair<int, int>, int, hash_pair> m;
        long long ans = 0;
        for(int i=0;i<rectangles.size();i++)
        {
            int x = rectangles[i][0];
            int y = rectangles[i][1];
            int g = gcd(x,y);
            x /= g;
            y /= g;
            ans += m[make_pair(x,y)];
            m[make_pair(x,y)]++;
        }
        return ans;
    }
};


// 解法3 
// unordered_map要想使用pair作为key，必须要实现一个hash
struct hash_pair { 
    template <class T1, class T2> 
    size_t operator()(const pair<T1, T2>& p) const
    { 
        auto hash1 = hash<T1>{}(p.first); 
        auto hash2 = hash<T2>{}(p.second); 
        return hash1 ^ hash2; 
    } 
}; 

class Solution {
public:
    int gcd(int a,int b)
    {
        if(b == 0)
            return a;
        return gcd(b,a%b);
    }
    long long interchangeableRectangles(vector<vector<int>>& rectangles) 
    {
        unordered_map<pair<int, int>, int, hash_pair> m;
        long long ans = 0;
        for(int i=0;i<rectangles.size();i++)
        {
            int x = rectangles[i][0];
            int y = rectangles[i][1];
            int g = gcd(x,y);
            x /= g;
            y /= g;
            ans += m[make_pair(x,y)];
            m[make_pair(x,y)]++;
        }
        return ans;
    }
};