【Leetcode】249. Group Shifted Strings

记录算法题解

已于 2024-01-05 16:54:57 修改

阅读量352

点赞数

分类专栏： LC 栈、队列、串及其他数据结构文章标签：字符串 leetcode hashmap 哈希 c++ 算法

于 2021-08-07 12:08:26 首次发布

本文链接：https://blog.csdn.net/qq_46105170/article/details/119483629

版权

LC 栈、队列、串及其他数据结构专栏收录该内容

678 篇文章

订阅专栏

该博客介绍了如何解决LeetCode上的字符串等价类问题，通过给定的C++代码实现将字符串数组按照等价关系进行归类。代码中使用unordered_map存储等价类，并以'a'开头的字符串作为代表元。时间复杂度为O(nl)，其中n是字符串数量，l是字符串平均长度。博客讨论了字符串操作和数据结构在解决算法问题中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

https://leetcode.com/problems/group-shifted-strings/

给定一个长 $n$ 的字符串数组 $A$ ，每个字符串只含英文小写字母。对于一个只含英文小写字母的字符串 $s$ ，可以将其每一个字母同时变成 $26$ 个字母中的后一个（'z'会变成'a'）。如果某个字符串可以通过这样的操作成为另一个字符串，则它们等价。求所有等价类。

规定以'a'开头的字符串是等价类的代表元，将每个字符串变为代表元就可以归类了。代码如下：

class Solution {
 public:
  vector<vector<string>> groupStrings(vector<string>& ss) {
    unordered_map<string, vector<string>> mp;
    for (auto& s : ss) {
      auto t = s;
      if (t[0] != 'a') {
        int diff = t[0] - 'a';
        for (int j = 0; j < t.size(); j++) {
          t[j] -= diff;
          if (t[j] < 'a') t[j] += 26;
        }
      }

      mp[t].push_back(s);
    }

    vector<vector<string>> res;
    for (auto& [k, v] : mp) res.push_back(v);
    return res;
  }
};

时空复杂度 $O (n l)$ 。