318. 最大单词长度乘积
给定一个字符串数组 words,找到 length(word[i]) * length(word[j]) 的最大值,并且这两个单词不含有公共字母。你可以认为每个单词只包含小写字母。如果不存在这样的两个单词,返回 0。
示例 1:
输入: [“abcw”,“baz”,“foo”,“bar”,“xtfn”,“abcdef”]
输出: 16
解释: 这两个单词为 “abcw”, “xtfn”。
示例 2:
输入: [“a”,“ab”,“abc”,“d”,“cd”,“bcd”,“abcd”]
输出: 4
解释: 这两个单词为 “ab”, “cd”。
示例 3:
输入: [“a”,“aa”,“aaa”,“aaaa”]
输出: 0
解释: 不存在这样的两个单词。
提示:
- 2 <= words.length <= 1000
- 1 <= words[i].length <= 1000
- words[i] 仅包含小写字母
方法一:位运算
class Solution {
public:
int maxProduct(vector<string>& words) {
// 生成掩码begin
vector<int> mask(words.size());
for(int i=0;i<words.size();i++){
string word = words[i];
for(int j=0;j<word.size();j++){
mask[i] |= 1<<(word[j]-'a');
}
}
// 生成掩码end
int maxValue = 0;
for(int i = 0; i < words.size(); i++){
for(int j = i + 1; j < words.size(); j++){
if((mask[i]&mask[j])==0){ //如果两个单词没有重复的字母
int lenMul = words[i].size()*words[j].size();
maxValue = lenMul>maxValue?lenMul:maxValue;
}
}
}
return maxValue;
}
};
每个单词有对应的掩码,如果两个单词之前没有重复,经过与运算后为0。
特别注意:很容易把(mask[i]&mask[j])==0
写成mask[i]&mask[j]==0
,需要注意,&的优先级比==低!
位运算的优化
我们注意到对于掩码位相同的字符串,我们只需要存最大的即可。
例如单词 meet 和 met 包含的字母相同,只是字母的出现次数和单词长度不同,因此这两个单词的位掩码表示也相同。计算最大长度时,单词 six和两个掩码相同的字符串,当然选择最大的那个meet。
class Solution {
public:
int maxProduct(vector<string>& words) {
unordered_map<int,int> map;
int length = words.size();
for (int i = 0; i < length; i++) {
int mask = 0;
string word = words[i];
int wordLength = word.size();
for (int j = 0; j < wordLength; j++) {
mask |= 1 << (word[j] - 'a');
}
map[mask] = wordLength>map[mask]?wordLength:map[mask];
}
int maxProd = 0;
for (auto [mask1, _] : map) {
int wordLength1 = map[mask1];
for (auto [mask2, _] : map) {
if ((mask1 & mask2) == 0) {
int wordLength2 = map[mask2];
maxProd = max(maxProd, wordLength1 * wordLength2);
}
}
}
return maxProd;
}
};
auto [mask1,_]
这是什么遍历方法!学到了!
时间复杂度:O(L + n2)。L为所有单词长度总和,n为单词数
空间复杂度:O(n)。其中n是单词数。