python leetcode 472. Concatenated Words

类似word break方法

class Solution(object):
    def findAllConcatenatedWordsInADict(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        def canBreak(w,wdict):
            if w=='':  flag[0]=True 
            for i in range(len(w)):
                if w[:i+1] in wdict and wdict[w[:i+1]]:
                    canBreak(w[i+1:],wdict)
                    if flag[0]: return True
            
        res=[]
        wdict={}
        for w in words:
            wdict[w]=True 
        for w in words:
            wdict[w]=False
            flag=[False]
            if canBreak(w,wdict):
                res.append(w)
            wdict[w]=True
        return res 

class Solution:
    def findAllConcatenatedWordsInADict(self, words):
        """
        :type words: List[str]
        :rtype: List[str]
        """
        def canBreak(w,wdict,mydict):
            if w in mydict: return mydict[w]
            if w=='':  return True 
            for i in range(len(w)):
                if w[:i+1] in wdict and wdict[w[:i+1]]:
                    if canBreak(w[i+1:],wdict,mydict):
                        mydict[w[:i+1]]=True
                        return True
            mydict[w[:i+1]]=False
            return False   
        res=[]
        wdict={}
        for w in words:
            wdict[w]=True 
        for w in words:
            wdict[w]=False
            flag=[False]
            if w!="" and canBreak(w,wdict,{}):
                res.append(w)
            wdict[w]=True
        return res