面试金典--面试题 17.15. 最长单词（递归）

题目描述

给定一组单词words，编写一个程序，找出其中的最长单词，且该单词由这组单词中的其他单词组合而成。若有多个长度相同的结果，返回其中字典序最小的一项，若没有符合要求的单词则返回空字符串。

示例：
输入： [“cat”,“banana”,“dog”,“nana”,“walk”,“walker”,“dogwalker”]
输出： “dogwalker”
解释： "dogwalker"可由"dog"和"walker"组成。

思路分析

首先上来就需要排序，因为题目要求找到其中最长的单词，所以按照长度进行排序，最长的在前面。长度相同则按照字典序排序。这样遍历的时候只要找到一个符合条件的答案就可以直接返回了。

实例：[“cat”,“banana”,“dog”,“nana”,“walk”,“walker”,“dogwalker”]
排序后：[‘dogwalker’, ‘banana’, ‘walker’, ‘nana’, ‘walk’, ‘cat’, ‘dog’]

大体思路：
按住一个单词，从剩下的单词里找是否能够组成该单词。
比如，按住dogwalker，遍历剩余的单词列表：[ ‘banana’, ‘walker’, ‘nana’, ‘walk’, ‘cat’, ‘dog’]。
剩余单词中，dog和walker可以组成dogwalker，所以直接返回dogwalker。

这个寻找的过程就直接用递归就行了。

# 递归外层
for i in range(len(words)):
    if dfs(words[i],words[i+1:]):
        print(words[i])

dfs(words[i],words[i+1:]) ：这里递归每次都是words[i+1:]。i+1因为是从长度长的单词开始遍历的，后续都是比他短的，短单词只能由比他更短的单词组成，所以这里直接往后遍历就行了。

# 递归体
def dfs(w,words):
    if not w:
        return True

    for i in range(len(words)):
        if words[i] == w[:len(words[i])]:
            if dfs(w[len(words[i]):],words):
                return True
    return False

完整代码

words = sorted(words,key=lambda x:(-len(x),x)) # 排序
print(words)
def dfs(w,words):
    if not w:
        return True
    for i in range(len(words)):
        if words[i] == w[:len(words[i])]:
            if dfs(w[len(words[i]):],words):
                return True
    return False

for i in range(len(words)):
    if dfs(words[i],words[i+1:]):
        return words[i]