【哈希表】A016_LC_猜字谜（状压 + dfs/位运算枚举子集）

一、Problem

外国友人仿照中国字谜设计了一个英文版猜字谜小游戏，请你来猜猜看吧。字谜的迷面 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] 所包含的字符都不重复

二、Solution

方法一：

这题关键是如何将 word 的贡献添加到 mask 中

思路

处理 words：预先将 words 中的各个 word 的贡献压缩表示到 int mask 中，用 map 记录各种 mask 出现的次数

处理 puzzles：一个 puzzles[i] 可能对应多个 word，但由于 puzzle 是固定顺序的，强硬对 puzzle 和 word 进行比对看你会导致一个 puzzle 能匹配的 word 变少；又因为 puzzle 长度最大为 7，故可以用 $2^n$ 算法枚举出 puzzle 的所有子集，再从 map 中找每个子集对应的状态可以匹配到的 word 的数量

class Solution {
public:
	void dfs(int i, int st, string& s, vector<int>& now) {
		if (i==s.size()) {
			now.emplace_back(st);
			return;
		}
		if (i>0) dfs(i+1, st, s, now);
		dfs(i+1, st|(1<<(s[i]-'a')), s, now);
	}
    vector<int> findNumOfValidWords(vector<string>& words, vector<string>& puzzles) {
    	unordered_map<int, int> cnt;
    	for (string& w : words) {
    		int mask=0;
    		for (char c : w)
    			mask |= 1<<(c-'a');
    		cnt[mask]++;
    	}
    	
   		int m=puzzles.size();
    	vector<int> ans(m);
    	for (int i=0; i<m; i++) {
    		vector<int> now;
            dfs(0, 0, puzzles[i], now);
    		int c=0;
    		for (int st : now) 
    			c+=cnt[st];
    		ans[i]=c;
    	}
    	return ans;
    }
};

446ms…

复杂度分析

• Time：$O(n)$，
• Space：$O(n)$，

---

方法二：利用位运算替换暴搜子集

参考别人的思路

利用 st & (st-1) 来去掉一个 1

class Solution {
public:
    vector<int> findNumOfValidWords(vector<string>& words, vector<string>& puzzles) {
    	unordered_map<int, int> cnt;
    	for (string& w : words) {
    		int mask=0;
    		for (char c : w)
    			mask |= 1<<(c-'a');
    		cnt[mask]++;
    	}

    	int m=puzzles.size();
    	vector<int> ans(m);
    	for (int i=0; i<m; i++) {
    		int S=0;
    		for (char c : puzzles[i])
    			S |= 1 << (c-'a');
    		for (int T=S; T; T=(T-1)&S) if (T & (1<<(puzzles[i][0]-'a'))) {
    			ans[i] += cnt[T];
            }
    	}
    	return ans;
    }
};

332ms…

复杂度分析

• Time：$O(n)$，
• Space：$O(n)$，