dfs 搜索

最新推荐文章于 2023-08-11 23:14:48 发布

**：天下第一

最新推荐文章于 2023-08-11 23:14:48 发布

阅读量182

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM

本文链接：https://blog.youkuaiyun.com/u013200703/article/details/93196589

ACM 专栏收录该内容

1 篇文章

订阅专栏

博客围绕DFS搜索展开，虽暂无具体内容，但DFS搜索是信息技术领域重要算法，常用于图和树的遍历等场景，能高效解决路径查找等问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution:
    """
    @param: words: a set of words without duplicates
    @return: all word squares
    """
    def wordSquares(self, words):
        if len(words) == 0:
            return []
        # write your code here
        # pre = set()
        pre = collections.defaultdict(set)
        n = len(words)
        m = len(words[0])
        for word in words:
            for i in range(m):
                pre[word[:i]].add(word)
                
        ans = []
        
        def search(word, line, matrix, has):
            # has.add(word)
            if len(matrix) == m:
                ans.append(matrix[:])
            else:
                prefix = "".join([matrix[i][line] for i in range(line)])
                for tar in pre[prefix]:
                    if tar not in has:
                        matrix.append(tar)
                        has.add(tar)
                        search(tar, line + 1, matrix, has)
                        has.remove(tar)
                        matrix.pop()
        
        for word in words:
            has = set([word])
            search(word, 1, [word], has)
        
        return ans