外國友人仿照中國字謎設計了一個英文版猜字謎小游戲,請你來猜猜看吧。
字謎的迷面 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’。
解題思路
將word單詞壓縮成int類型,通過枚舉puzzle的子集,匹配對應的word
代碼
class Solution {public List<Integer> findNumOfValidWords(String[] words, String[] puzzles) {Map<Integer,Integer> map=new HashMap<>();for (String word : words) {//壓縮word,并且記錄相同的出現次數int mask=0;for (int i = 0; i < word.length(); i++) {mask|=(1<<(word.charAt(i)-'a'));}if(Integer.bitCount(mask)<=7)map.put(mask,map.getOrDefault(mask,0)+1);}List<Integer> list=new ArrayList<>();for (String puzzle : puzzles) {int cur=0;int mask=0;for (int i = 1; i < 7; i++) {mask|=(1<<(puzzle.charAt(i)-'a'));}int sub=mask;do{//枚舉二進制子集的代碼int s=sub|(1<<(puzzle.charAt(0)-'a'));//保證第一位是固定的if(map.containsKey(s)) cur+=map.get(s);//存在符合的,累加到當前結果sub=(sub-1)&mask;}while (sub!=mask);list.add(cur);}return list;}
}