LeetCode 题解系列(1160)：拼写单词

题目描述:

给你一份『词汇表』（字符串数组） words 和一张『字母表』（字符串） chars。

假如你可以用 chars 中的『字母』（字符）拼写出 words 中的某个『单词』（字符串），那么我们就认为你掌握了这个单词。

注意：每次拼写（指拼写词汇表中的一个单词）时，chars 中的每个字母都只能用一次。

返回词汇表 words 中你掌握的所有单词的 长度之和。

 

示例 1：

输入：words = [“cat”,”bt”,”hat”,”tree”], chars = “atach”
输出：6
解释：
可以形成字符串 “cat” 和 “hat”，所以答案是 3 + 3 = 6。
示例 2：

输入：words = [“hello”,”world”,”leetcode”], chars = “welldonehoneyr”
输出：10
解释：
可以形成字符串 “hello” 和 “world”，所以答案是 5 + 5 = 10。

解法一: bit map

1. 使用 bit map 统计每个单词的字符词频。
2. 遍历每个单词，判断这个单词能否由这个词频的单词组成。

时间复杂度: O(N); 辅助空间复杂度: O(N)

class Solution {
    public int countCharacters(String[] words, String chars) {
        if (chars.length() == 0) {
            return 0;
        }
        int[] count = new int[26];
        for (int i = 0; i < chars.length(); i++) {
            count[chars.charAt(i) - 'a']++;
        }
        int res = 0;
        for (int i = 0; i < words.length; i++) {
            res += learn(count, words[i]);
        }
        return res;
    }

    private int learn(int[] count, String word) {
        int i = 0;
        // 统计能否满足掌握这个单词
        for (; i < word.length(); i++) {
            if (--count[word.charAt(i) - 'a'] < 0) {
                break;
            }
        }
        // 复原 count
        for (int j = 0; j <= Math.min(i, word.length() - 1); j++) {
            count[word.charAt(j) - 'a']++;
        }
        return i == word.length() ? word.length() : 0;
    }
}