力扣（LeetCode）算法_C++——移位字符串分组

吴俊荣2023-09-07 19:05

给定一个字符串，对该字符串可以进行 "移位" 的操作，也就是将字符串中每个字母都变为其在字母表中后续的字母，比如："abc" -> "bcd"。这样，我们可以持续进行 "移位" 操作，从而生成如下移位序列：

"abc" -> "bcd" -> ... -> "xyz"

给定一个包含仅小写字母字符串的列表，将该列表中所有满足 "移位" 操作规律的组合进行分组并返回。

示例：

输入： $"abc", "bcd", "acef", "xyz", "az", "ba", "a", "z"$

输出：

\["abc","bcd","xyz"\], \["az","ba"\], \["acef"\], \["a","z"

]

解释：可以认为字母表首尾相接，所以 'z' 的后续为 'a'，所以 $"az","ba"$ 也满足 "移位" 操作规律。

cpp 复制代码

    vector<vector<string>> groupStrings(vector<string>& strings) {
        unordered_map<string, vector<string>> mp;
        vector<vector<string>> ans;

        for (auto& str: strings)
        {
            string key(str);
            for (auto& c: key)
            {
                c = (c - str[0] + 26) % 26 + 'a';
            }
            mp[key].push_back(str);
        }

        
        for (auto& m: mp)
        {
            ans.push_back(m.second);
        }

        return ans;
    }