题目
剑指 Offer II 005. 单词长度的最大乘积
给定一个字符串数组 words,请计算当两个字符串 words[i] 和 words[j] 不包含相同字符时,它们长度的乘积的最大值。假设字符串中只包含英语的小写字母。如果没有不包含相同字符的一对字符串,返回 0。
原题链接: 【点此查看】
示例1:
输入: words = ["abcw","baz","foo","bar","fxyz","abcdef"]
输出: 16
解释: 这两个单词为 "abcw", "fxyz"。它们不包含相同字符,且长度的乘积最大。
示例2:
输入: words = ["a","aa","aaa","aaaa"]
输出: 0
解释: 不存在这样的两个单词。
分析
将判断两个单词是否有公共字母的时间复杂度降低到 O(1),则可以将总时间复杂度降低到 O(n^2)。可以使用位运算预处理每个单词,通过位运算操作判断两个单词是否有公共字母。
由于单词只包含小写字母,共有 26 个小写字母,因此可以使用位掩码的最低 26 位分别表示每个字母是否在这个单词中出现。将 a 到 z 分别记为第 0 个字母到第 25 个字母,则位掩码的从低到高的第 i 位是 1 当且仅当第 i 个字母在这个单词中,其中 0 ≤ i ≤ 25。
题解
class Solution {
public:
int maxProduct(vector<string>& words) {
int n = words.size();
if (n <= 0) return 0;
vector<int> mask(n);
for (int i = 0; i < n; ++i)
{
string word = words[i];
int wordLength = word.size();
for (int j = 0; j < wordLength; ++j)
{
mask[i] |= 1 << (word[j] - 'a');
}
}
int maxProd = 0;
for (int i = 0; i < n; ++i)
{
for (int j = i + 1; j < n; ++j)
{
if ((mask[i] & mask[j]) == 0)
maxProd = std::max(maxProd, static_cast<int>(words[i].size() * words[j].size()));
}
}
return maxProd;
}
};