472. Concatenated Words leetcode

findAllConcatenatedWordsInADict.go

func findAllConcatenatedWordsInADict(words []string) []string {
    res:= []string{}
    mp:= make(map[string]int)
    
    for _, word := range words {
        mp[word] = 1
    }
    
    for _, word := range words {
        delete(mp, word)
        
        c:=breakWord(word, mp)
        
        if c >1 {
            res = append(res, word)
        } else if c == 0 {
            mp[word] = 1
        }
    }
    return res
}        

func breakWord(word string, mp map[string]int) int {
    if _, ok := mp[word]; ok {
        return mp[word]
    }
    for i:=1; i < len(word); i++ {
        str:= word[0:i]
        _, ok := mp[str]
        if ok && mp[str] > 0 {
            c1 := mp[str]
            c2 := breakWord(word[i:], mp)
            
            if c2 > 0 {
                mp[word] = c1 + c2
                return c1 + c2
            }
        }
    }   
    mp[word] = 0
    return 0
}