力扣 1178. 猜字谜 位运算 思维

https://leetcode-cn.com/problems/number-of-valid-words-for-each-puzzle/
在这里插入图片描述
思路:看完题目会发现关键点是一个单词中出现的字符种类,和个数没有关系,那么我们可以用位运算表示某个单词含有的字符种类。具体做法是,如果它含有小写字母 x x x,我们就可以把它二进制表示的第 x − ′ a ′ x-'a' xa位置为 1 1 1。这样就将字符串变成了整数,接下来构建一个哈希表,记录某个整数所对应的字符串个数。对于每个 p u z z l e puzzle puzzle我们可以得到它的哈希值,接下来就是枚举它的所有子集,将对应的字符串个数累加起来即可。

class Solution {
    
    
public:
    vector<int> findNumOfValidWords(vector<string>& words, vector<string>& puzzles) {
    
    
        unordered_map<int,int> m;
        int strHashVal=0;
        for(const string &s : words)
        {
    
    
            strHashVal=0;
            for(char c : s)
                strHashVal|=1<<(c-'a');
            if(__builtin_popcount(strHashVal)<=7)
                ++m[strHashVal];
        }
        int subset=0,n=puzzles.size(),cnt=0;
        auto mend=m.end();
        vector<int> ans(n);
        int i=0;
        for(const string &s : puzzles)
        {
    
    
            strHashVal=cnt=0;
            for(int j=1;j<7;j++)
                strHashVal|=1<<(s[j]-'a');
            subset=strHashVal;
            do
            {
    
    
                int tmp=subset | (1<<(s[0]-'a'));
                auto it=m.find(tmp);
                if(it!=mend)
                    cnt+=it->second;
                subset=(subset-1)&strHashVal;
            }while(subset!=strHashVal);
            ans[i++]=cnt;
        }
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/xiji333/article/details/114221672