【哈希表, 位运算, 状态压缩】统计一致字符串的数目

题目

1684. 统计一致字符串的数目

---

给你一个由不同字符组成的字符串 allowed 和一个字符串数组 words 。如果一个字符串的每一个字符都在 allowed 中，就称这个字符串是 一致字符串 。

请你返回 words 数组中 一致字符串 的数目。

示例 1：

输入：allowed = "ab", words = ["ad","bd","aaab","baa","badab"]
输出：2
解释：字符串 "aaab" 和 "baa" 都是一致字符串，因为它们只包含字符 'a' 和 'b' 。

示例 2：

输入：allowed = "abc", words = ["a","b","c","ab","ac","bc","abc"]
输出：7
解释：所有字符串都是一致的。

示例 3：

输入：allowed = "cad", words = ["cc","acd","b","ba","bac","bad","ac","d"]
输出：4
解释：字符串 "cc"，"acd"，"ac" 和 "d" 是一致字符串。

提示：

• 1 <= words.length <= 10^4
• 1 <= allowed.length <= 26
• 1 <= words[i].length <= 10
• allowed 中的字符 互不相同 。
• words[i] 和 allowed 只包含小写英文字母。

解题

方法一：哈希表

思路

把 allowed 中所有的字符放入哈希表(has)中，随后遍历 words，对于每一个字符串，如果其中出现了未在 has 中出现的字符就跳过，否则计数(ans)增加，最后返回计数即可。

代码

class Solution {
    public int countConsistentStrings(String allowed, String[] words) {
        boolean[] has = new boolean[26];
        for (char ch : allowed.toCharArray()) has[ch - 'a'] = true;
        int ans = 0;
        outer: for (String word : words) {
            for (char ch : word.toCharArray()) {
                if (!has[ch - 'a']) continue outer;
            }
            ++ans;
        }
        return ans;
    }
}

优化

字符串中只会出现小写英文字母，所有可以把状态压缩到 $26$ 位中：

class Solution {
    public int countConsistentStrings(String allowed, String[] words) {
        int has = 0;
        for (char ch : allowed.toCharArray()) has |= 1 << ch - 'a';
        int ans = 0;
        outer: for (String word : words) {
            for (char ch : word.toCharArray()) {
                if (((has >> ch - 'a') & 1) == 0) continue outer;
            }
            ++ans;
        }
        return ans;
    }
}