题目要求
字谜的迷面 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’。
代码
class Solution {
public:
int CountOnes(int n){ //count ones of binary number n
int count=0;
while(n){
count++;
n=n&(n-1);
}
return count;
}
vector<int> findNumOfValidWords(vector<string>& words, vector<string>& puzzles) {
unordered_map<int,int> word_bitmap; //key:word->bitmask,value:number of occurrences
int bitmask;
for(string word:words){
bitmask=0;
for(char ch:word){
bitmask|=1<<(ch-'a');
}
if(CountOnes(bitmask)<=7){
word_bitmap[bitmask]++;
}
}
vector<int> answer;
int cnt;
for(string puzzle:puzzles){
bitmask=0,cnt=0;
for(char ch:puzzle){
bitmask|=1<<(ch-'a');
}
// check subsets of puzzle's bitmask and add up numbers of occurrences
for(int subset=bitmask;subset;subset=(subset-1)&bitmask){
if((subset&(1<<(puzzle[0]-'a')))&&word_bitmap.count(subset)){
cnt+=word_bitmap[subset];
}
}
answer.push_back(cnt);
}
return answer;
}
};
笔记
- 遍历map:for(auto &ite:map){ cout<<ite.first<<ite.second<<endl;}
- 枚举二进制子集:伪代码:for(int i=s; i ; i=(i-1)&s)
主要意思:当i不为0时,i-1表示将i最后一位1及其后面全部取反,(i-1)&s表示去掉i最后一位1 - map和unordered_map:map是红黑树实现,较为有序。unordered_map是哈希表实现,当数据较多且查找次数多时,性能更加优秀。