unordered_map 自定义结构为Key

Posted 2021-10-29 顾文繁

有如下结构体

struct Word {
	int len = 0;
	int ascii = 0;
	Word(string str)
	{
		len = str.length();
		for (char c : str)
		{
			ascii += c;
		}
	}
};

需要使用如下场景

unordered_map<Word, vector<string>>m;

这里需要实现哈希函数和重载()函数

struct Hash
{
	std::size_t operator()(const Word& word) const
	{
		using std::size_t;
		using std::hash;

		return ((hash<int>()(word.len)
			^ (hash<int>()(word.ascii) << 1)) >> 1);
	}
};

struct Equal
{
	bool operator () (const Word& lhs, const Word& rhs) const
	{
		return lhs.len == rhs.len
			&& lhs.ascii == rhs.ascii;
	}
};

然后就可以使用ordered_map了

unordered_map<Word, vector<string>, Hash, Equal> m;
		vector<vector<string>> ret;
		for (auto& str : strs)
		{
			Word w(str);
			auto iter = m.find(w);
			if (iter == m.end())
			{
				vector<string> v;
				v.push_back(str);
				m.insert(pair<Word, vector<string>>(w, v));
			}
			else
			{
				iter->second.push_back(str);
			}
		}

		for (auto it = m.begin(); it != m.end(); it++)
		{
			ret.push_back(it->second);
		}