【算法练习】leetcode 二进制子集 1178. 猜字谜/78. 子集

1178. 猜字谜

难度困难149

外国友人仿照中国字谜设计了一个英文版猜字谜小游戏，请你来猜猜看吧。

字谜的迷面 puzzle 按字符串形式给出，如果一个单词 word 符合下面两个条件，那么它就可以算作谜底：

• 单词 word 中包含谜面 puzzle 的第一个字母。
• 单词 word 中的每一个字母都可以在谜面 puzzle 中找到。
  例如，如果字谜的谜面是 "abcdefg"，那么可以作为谜底的单词有 "faced", "cabbage", 和 "baggage"；而 "beefed"（不含字母 "a"）以及 "based"（其中的 "s" 没有出现在谜面中）都不能作为谜底。

返回一个答案数组 answer，数组中的每个元素 answer[i] 是在给出的单词列表 words 中可以作为字谜迷面 puzzles[i] 所对应的谜底的单词数目。

 

示例：

输入：
words = ["aaaa","asas","able","ability","actt","actor","access"], 
puzzles = ["aboveyz","abrodyz","abslute","absoryz","actresz","gaswxyz"]
输出：[1,1,3,2,4,0]
解释：
1 个单词可以作为 "aboveyz" 的谜底 : "aaaa" 
1 个单词可以作为 "abrodyz" 的谜底 : "aaaa"
3 个单词可以作为 "abslute" 的谜底 : "aaaa", "asas", "able"
2 个单词可以作为 "absoryz" 的谜底 : "aaaa", "asas"
4 个单词可以作为 "actresz" 的谜底 : "aaaa", "asas", "actt", "access"
没有单词可以作为 "gaswxyz" 的谜底，因为列表中的单词都不含字母 'g'。

 

提示：

• 1 <= words.length <= 10^5
• 4 <= words[i].length <= 50
• 1 <= puzzles.length <= 10^4
• puzzles[i].length == 7
• words[i][j], puzzles[i][j] 都是小写英文字母。
• 每个 puzzles[i] 所包含的字符都不重复。

 

解题思路

首先将word全部用二进制表示

如abd 表示成   d c b a

                        1 0 1 1

然后对于每一个谜语 puzzle 枚举所有的子集 但是值得注意的是 枚举的时候要枚举 一定要包含第一个字母的子集

 先把puzzle用二进制表示

subset=..

mask=subset

 

do{

int s=subset & (1<<puzzle[0]);

if(freq.count(s)){

    

}

subset=mask&(subset-1);

}while(subset!=mask)

 

关键在于如何枚举判断一个二进制状态数字k的子集, 方法就是针对中的二进制为1的位开始进行减法，判断数字k的二进制子集, 像枚举(2^k-1) ~ 0一样枚举其子集

    int sub = k;
    do {
        sub = (sub - 1) & k;
    } while(sub != k);

作者：huanggangfeng
链接：https://leetcode-cn.com/problems/number-of-valid-words-for-each-puzzle/solution/zhi-dao-liao-ru-he-mei-ju-yi-ge-er-jin-z-2rph/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

 

代码

class Solution {
public:
 vector<int> findNumOfValidWords(vector<string>& words, vector<string>& puzzles) {
        unordered_map<int, int> frequency;
        //单词 用二进制表示 abd ->1011
        vector<int> ans;
        for(string word:words){
            int mask=0;
            for(int i=0;i<word.size();i++){
                mask|=(1<<(word[i]-'a'));
            }
            frequency[mask]++;  //mask是word的二进制表示
        }
        for(string puzzle:puzzles){
            int mask=0;
            for(int i=0;i<puzzle.size();i++){
                mask|=(1<<(puzzle[i]-'a'));
            }
            int subset=mask;
            int total=0;
            do{
                if(subset & (1<<(puzzle[0]-'a'))){
                    //如果包含puzzle的首字母
                    total+=frequency[subset];
                }
                subset=(subset-1)&mask;
            }while(subset!=mask);
            ans.push_back(total);
        }
        return ans;
    }
};