LeetCode C++30-字母异位词分组

题目描述

        leetcode原题链接：字母异位词分组 

        给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。

字母异位词 是由重新排列源单词的字母得到的一个新单词，所有源单词中的字母通常恰好只用一次。

举例

示例 1:   输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
              输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
示例 2:   输入: strs = [""]         输出: [[""]]
示例 3:   输入: strs = ["a"]       输出: [["a"]]

解题方法：由于都是字母，可以将每一个字符串映射到长度为26的int数组中，每个字符出现的个数作为对应值。例如：ababc映射为26位数字的值为：221000...000(后面23个0），这样，异位词就会映射为同一个26位数字。

代码

#include <iostream>
#include <string>
#include <vector>
#include <map>

std::string str2alphabet(const std::string& str) {
	std::string result(26, '0');
	int n = str.size();
	for (int i = 0; i < n; i++) {
		int index = str[i] - 'a';// 每个字符对应的26位空间的位置
		int cnt = result[index];//先获取当前字符的个数
		cnt++;
		result[index] = '0' + cnt;
	}
	return result;
}

std::vector<std::vector<std::string>> groupAnagrams(std::vector<std::string>& strs) {
    std::vector<std::vector<std::string>> result;
	std::map<std::string, int> str2index;//str26对应的vector index
	for (auto& str : strs) {
		std::string str26 = str2alphabet(str);//先转换为26位空间字符串
		if (str2index.count(str26) > 0) {
			// 先获取对应的下标
			int index = str2index[str26];
			result[index].push_back(str);
		} else {//首次扫描到
			int size = result.size();//当前的结果分组组数
			//新插入的元素对应的下标是size，前一个是size-1
			str2index[str26] = size;
			result.push_back({str});//第size个result
		}
	}
	return result;
}

void print_result(const std::vector<std::string>& strs, 
				  const std::vector<std::vector<std::string>>& result) {
	std::cout << "========" << std::endl;
	std::cout << "input:" << std::endl;
	for (auto& str : strs) {
		std::cout << str << " ";
	}
	std::cout << std::endl;
	std::cout << "output:" << std::endl;
	for (auto& str_group : result) {
		for (auto& str : str_group) {
			std::cout << str << " ";
		}
		std::cout << std::endl;
	}
}

int main()
{
	// case1 : strs=["eat", "tea", "tan", "ate", "nat", "bat"]
	//   输出:[["bat"],["nat","tan"],["ate","eat","tea"]
	std::vector<std::string> strs = {"eat", "tea", "tan", "ate", "nat", "bat"};
	auto result = groupAnagrams(strs);
	print_result(strs, result);
	// case2: strs=[""] ==> [[""]]
	strs = {""};
	result = groupAnagrams(strs);
	print_result(strs, result);
	// case3: strs=["a"] ==> [["a"]]
	strs = {"a"};
	result = groupAnagrams(strs);
	print_result(strs, result);
    return 0;
}

代码运行结果如下：

========
input:
eat tea tan ate nat bat 
output:
eat tea ate 
tan nat 
bat 
========
input:
 
output:
 
========
input:
a 
output:
a