搜索树/特里时返回多个节点?

时间:2012-03-17 11:23:50

标签: c# tree suffix-tree

我构造了一个后缀trie,一个包含字符串所有后缀的树,其中每个节点只包含一个字符,每个路径末尾都有一个SuffixNode,它包含字符串中后缀的位置。

假设我的trie包含单词“Cat”,“Car”和“Can”,我想搜索“Ca”,结果应返回3个后缀节点,因为搜索字符串位于3个不同的位置。我设法在树中搜索“Ca”,但是一旦我到达那一点,我就不知道如何继续遍历'a'节点的子节点来查找所有后缀节点。

我正在考虑使用某种集合来添加后缀节点,然后返回集合。这种方法是否有意义,还是有更好的解决方案?

我已经解决了下面的搜索问题。它没有返回任何节点的原因与树的创建和节点之间的差异有关:

public void SearchTrie(Node parent, String s, List<SuffixNode> suffixes)
    {
        Node n = FindNode(parent, s);
        FindSuffixes(n, suffixes);
    }

    private void FindSuffixes(Node parent,List<SuffixNode> suffixes)
    {            
        if (parent is SuffixNode)
        {
            suffixes.Add((SuffixNode)parent);
        }
        else
        {
            foreach (KeyValuePair<char, Node> child in parent.children)
            {
                FindSuffixes(child.Value, suffixes);
            }
        }
    }

    private Node FindNode(Node parent, String s)
    {
        if ((s.Length == 1) && parent.children.ContainsKey(s[0]))
        {
            Node n = parent.children[s[0]];
            return n;
        }

        while (s[0].ToString() != "")
        {
            if (parent.children.ContainsKey(s[0]))
            {
                if ((s.Length > 1))
                {
                    return FindNode(parent.children[s[0]], s.Substring(1));
                }

            }
            else
            {
                break;
            }

        }

        return null;
    }

节点:

class Node
{
    public char label;
    public Node parent;
    public Dictionary<char,Node> children;

    public Node(Node NewParent, char NewLabel)
    {
        this.parent = NewParent;
        this.label = NewLabel;
        children=new Dictionary<char,Node>();
    }
}

1 个答案:

答案 0 :(得分:3)

  

我正在考虑使用某种集合来添加后缀节点,然后返回集合。

那将是第一选择。

  

......,还是有更好的解决方案?

还有其他解决方案,例如

  • 填写来电者提供的清单
  • 使用带yield return
  • 的迭代器块

但是没有任何特殊要求,请填写List<string>并将其作为IEnumerable<string>返回。