LeetCode 每日一题 Day 46 ||枚举

2744. 最大字符串配对数目

给你一个下标从 0 开始的数组 words ，数组中包含 互不相同 的字符串。

如果字符串 words[i] 与字符串 words[j] 满足以下条件，我们称它们可以匹配：

字符串 words[i] 等于 words[j] 的反转字符串。
0 <= i < j < words.length
请你返回数组 words 中的 最大 匹配数目。

注意，每个字符串最多匹配一次。

示例 1：

输入：words = [“cd”,“ac”,“dc”,“ca”,“zz”]
输出：2
解释：在此示例中，我们可以通过以下方式匹配 2 对字符串：

• 我们将第 0 个字符串与第 2 个字符串匹配，因为 word[0] 的反转字符串是 “dc” 并且等于 words[2]。
  -> 我们将第 1 个字符串与第 3 个字符串匹配，因为 word[1] 的反转字符串是 “ca” 并且等于 words[3]。
  可以证明最多匹配数目是 2 。

示例 2：

输入：words = [“ab”,“ba”,“cc”]
输出：1
解释：在此示例中，我们可以通过以下方式匹配 1 对字符串：

• 我们将第 0 个字符串与第 1 个字符串匹配，因为 words[1] 的反转字符串 “ab” 与 words[0] 相等。
  可以证明最多匹配数目是 1 。

示例 3：

输入：words = [“aa”,“ab”]
输出：0
解释：这个例子中，无法匹配任何字符串。

提示：

1 <= words.length <= 50
words[i].length == 2
words 包含的字符串互不相同。
words[i] 只包含小写英文字母。

根据题意枚举即可：

class Solution {
public:
    int maximumNumberOfStringPairs(vector<string>& words) {
        int n = words.size();
        int res= 0;
        for(int i = 0;i < n;i++){
            for(int j = i+1;j < n;j++){
                if(words[i][0] == words[j][1] && words[i][1] == words[j][0]){
                    res++;
                }
            }
        }
        return res;
    }
};

才发现评论里有大佬指正复杂度问题，确实上面的代码复杂度为n方，通过哈希优化可以到On：

class Solution {
public:
    int maximumNumberOfStringPairs(vector<string>& words) {
        unordered_map<string, int> counter;
        int res = 0;
        for (const string& word : words) {
            string key = word.substr(0, 1) + word.substr(1, 1);
            string reversed_key = string(key.rbegin(), key.rend());
            if (counter.count(reversed_key)) {
                res += counter[reversed_key];
                counter[reversed_key]--;
                if (counter[reversed_key] == 0) {
                    counter.erase(reversed_key);
                }
            } else {
                counter[key]++;
            }
        }
        return res;
    }
};