迭代地组合链表

时间:2016-03-07 01:34:06

标签: python linked-list iteration

我正在尝试迭代地组合两个链接列表,但我现在所拥有的是给出了与结果相反的结果。有没有办法以正确的顺序组合列表,而不必在创建结果列表后反转?

class Link: 
    empty = ()

    def __init__(self, first, rest=empty):
        assert rest is Link.empty or isinstance(rest, Link)
        self.first = first
        self.rest = rest

    def __add__(self, lst):
        """ 
        >>>s = Link(1, Link(2))
        >>>s + Link(3,Link(4))
        Link(1,Link(2,Link(3,Link(4))))
        """
        result = Link.empty 
        while self is not Link.empty: 
            result = Link(self.first,result)
            self = self.rest 

        while lst is not Link.empty: 
            result = Link(lst.first, result)
            lst = lst.rest 
        return result 

2 个答案:

答案 0 :(得分:0)

避免更改self,找到第一个列表的末尾,并将第二个列表添加到其中。

def __add__(self, lst):
    current = self

    while current.rest != self.empty:
        current = current.rest

    current.rest = lst

    return self

或者,如果您希望返回新的链接列表:

def __add__(self, lst):
    new_list = Link(self.first)
    new_link = new_list

    current = self

    while current.rest != self.empty:
        new_link.rest = Link(current.first)
        new_link = new_link.rest

        current = current.rest

    current = lst

    while current != self.empty:
        new_link.rest = Link(current.first)
        new_link = new_link.rest

        current = current.rest

    return new_list

注意:这将引用每个Link.first值(不是独立副本)。

答案 1 :(得分:0)

从概念上讲,要合并两个链接列表,您需要做的就是找到第一个列表的尾部,并将其与第二个列表的 head 连接,然后返回第一个列表的头部。当前代码的主要问题是你没有留在第一个列表的头部。

我假设您希望__add__方法创建由self后跟lst组成的副本,因此只需创建Link的副本即可首先来自self然后lst,并在迭代时附加它们。像这样:

def __add__(self, lst):
    result = Link(self.first)
    cur = result
    self = self.rest
    # Copy our list
    while self is not Link.empty:
        cur.rest = Link(self.first)
        cur = cur.rest
        self = self.rest
    # Copy and connect the Links in lst
    while lst is not Link.empty:
        cur.rest = Link(lst.first)
        cur = cur.rest
        lst = lst.rest
    return result

要诊断当前代码的错误,请考虑单步执行示例。假设selfLink(1,Link(2))

    result = Link.empty 
    while self is not Link.empty: 
        result = Link(self.first,result)

现在result是什么?

  • Link(self.first,result)
  • Link(1, Link.empty)

        self = self.rest
    

现在selfLink(2,())

        result = Link(self.first,result)

现在result是什么?

  • Link(self.first,result)
  • Link(2, Link(1, Link.empty))

糟糕。发现问题;你正在反向连接链接。