LeetCode-Python-1255. 得分最高的单词集合（回溯）

你将会得到一份单词表 words，一个字母表 letters （可能会有重复字母），以及每个字母对应的得分情况表 score。

请你帮忙计算玩家在单词拼写游戏中所能获得的「最高得分」：能够由 letters 里的字母拼写出的 任意 属于 words 单词子集中，分数最高的单词集合的得分。

单词拼写游戏的规则概述如下：

• 玩家需要用字母表 letters 里的字母来拼写单词表 words 中的单词。
• 可以只使用字母表 letters 中的部分字母，但是每个字母最多被使用一次。
• 单词表 words 中每个单词只能计分（使用）一次。
• 根据字母得分情况表score，字母 'a', 'b', 'c', ... , 'z' 对应的得分分别为 score[0], score[1], ..., score[25]。
• 本场游戏的「得分」是指：玩家所拼写出的单词集合里包含的所有字母的得分之和。

 

示例 1：

输入：words = ["dog","cat","dad","good"], letters = ["a","a","c","d","d","d","g","o","o"], score = [1,0,9,5,0,0,3,0,0,0,0,0,0,0,2,0,0,0,0,0,0,0,0,0,0,0]
输出：23
解释：
字母得分为  a=1, c=9, d=5, g=3, o=2
使用给定的字母表 letters，我们可以拼写单词 "dad" (5+1+5)和 "good" (3+2+2+5)，得分为 23 。
而单词 "dad" 和 "dog" 只能得到 21 分。

示例 2：

输入：words = ["xxxz","ax","bx","cx"], letters = ["z","a","b","c","x","x","x"], score = [4,4,4,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,5,0,10]
输出：27
解释：
字母得分为  a=4, b=4, c=4, x=5, z=10
使用给定的字母表 letters，我们可以组成单词 "ax" (4+5)， "bx" (4+5) 和 "cx" (4+5) ，总得分为 27 。
单词 "xxxz" 的得分仅为 25 。

示例 3：

输入：words = ["leetcode"], letters = ["l","e","t","c","o","d"], score = [0,0,1,1,1,0,0,0,0,0,0,1,0,0,1,0,0,0,0,1,0,0,0,0,0,0]
输出：0
解释：
字母 "e" 在字母表 letters 中只出现了一次，所以无法组成单词表 words 中的单词。

 

提示：

• 1 <= words.length <= 14
• 1 <= words[i].length <= 15
• 1 <= letters.length <= 100
• letters[i].length == 1
• score.length == 26
• 0 <= score[i] <= 10
• words[i] 和 letters[i] 只包含小写的英文字母。

思路：

数据规模暗示回溯，题意明示回溯。

只要能找到所有可能的组合，在所有组合的得分中取最大值返回即可。

所以是一道常规回溯题，具体细节看代码。

时间复杂度：O(2 ^ len(words)) * max(words[i].length))

空间复杂度：O（26 + 26 + max(words[i].length)) 

class Solution(object):
    def maxScoreWords(self, words, letters, score):
        """
        :type words: List[str]
        :type letters: List[str]
        :type score: List[int]
        :rtype: int
        """
        from collections import defaultdict, Counter
        dic = dict()
        letter_dic = defaultdict(int)
        for i, val in enumerate(score):#构建一个字典
            dic[chr(ord("a") + i)] = val #key是 字母，val是字母对应的分数
            
        letter_dic = Counter(letters)#构建另一个字典， key是字母， val是每次字母剩余的个数
 
        s = set(letters)
        v_words = []
        for word in words:#删掉所有根本不可能被构成的单词
            flag = 0
            for char in word:
                if char not in s:
                    flag = 1
            if flag: # 如果一个单词里存在某个在letters里找不到的字母，则无需考虑这个单词
                continue
            v_words.append(word)
        self.res = 0
                
        def helper(word, letter_dic):
            # return True 如果word能用letter_dic里的letter构成，否则返回False
            dicc = collections.Counter(word)
            for key in dicc:
                if dicc[key] > letter_dic[key]:
                    return False
            return True
        
        def dfs(start, tmp):
            self.res = max(self.res, tmp)
            if start >= len(v_words):
                return
            
            for i in range(start, len(v_words)):#从start开始找，避免重复
                if helper(v_words[i], letter_dic):#如果当前单词可以被构成
                    for char in v_words[i]: #构成它，更新字典
                        letter_dic[char] -= 1
                    dfs(i + 1, tmp + sum([dic[char] for char in v_words[i]])) #dfs下一层
                    for char in v_words[i]: #回溯，复原所有状态
                        letter_dic[char] += 1                   
        dfs(0, 0)
        return self.res