Description
We are given two arrays A and B of words. Each word is a string of lowercase letters.
Now, say that word b is a subset of word a if every letter in b occurs in a, including multiplicity. For example, “wrr” is a subset of “warrior”, but is not a subset of “world”.
Now say a word a from A is universal if for every b in B, b is a subset of a.
Return a list of all universal words in A. You can return the words in any order.
题目链接:https://leetcode.com/problems/word-subsets/description/
Difficulty: hard
Example 1:
Input: A = ["amazon","apple","facebook","google","leetcode"], B = ["e","o"]
Output: ["facebook","google","leetcode"]
Example 2:
Input: A = ["amazon","apple","facebook","google","leetcode"], B = ["l","e"]
Output: ["apple","google","leetcode"]
Example 3:
Input: A = ["amazon","apple","facebook","google","leetcode"], B = ["e","oo"]
Output: ["facebook","google"]
Example 4:
Input: A = ["amazon","apple","facebook","google","leetcode"], B = ["lo","eo"]
Output: ["google","leetcode"]
Example 5:
Input: A = ["amazon","apple","facebook","google","leetcode"], B = ["ec","oc","ceo"]
Output: ["facebook","leetcode"]
Note:
- 1 <= A.length, B.length <= 10000
- 1 <= A[i].length, B[i].length <= 10
- A[i] and B[i] consist only of lowercase letters.
- All words in A[i] are unique: there isn’t i != j with A[i] == A[j].
分析
- 如题所定义的包含关系可以转化为,字符串A,B,若B中的每一个字母出现的次数都小于等于A中字母出现的次数,则A包含B;
- 对于B中每一个字符串,统计对应字母出现次数最多的次数,存在dicb中;
- 对于A中每个字符串a,计算a中每个字母出现的次数,存在dica中;
- 若dica包含dicb,则a是满足要求的字符串。
参考代码
class Solution:
def wordSubsets(self, A, B):
    from collections import defaultdict
    dicb=defaultdict(lambda:0)
    for b in B:
        d=defaultdict(lambda:0)
        for tar in b:
            d[tar]+=1
        for tar in d:
            dicb[tar]=max(dicb[tar],d[tar])
    li=[]
    #print(dicb)
    for a in A:
        dica=defaultdict(lambda:0)
        for tar in a:
            dica[tar]+=1
        #print(dica)
        if(all(dica[c]>=dicb[c] for c in dicb)):
            li.append(a)
    return li