面试金典--面试题 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]

你可能感兴趣的:(python,深度优先,算法)