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;
    }
};

猜你喜欢

转载自blog.csdn.net/m0_73814009/article/details/135659373