python--Python list转为链表

代码即注释如下

import sys
import json


class ListNode:
    """
    定义一个节点,包括value 和next指针,指向下一个node
    """

    def __init__(self, x):
        self.val = x
        self.next = None


def stringToIntegerList(input):
    return json.loads(input)


def stringToListNode(input):
    # Generate list from the input
    numbers = stringToIntegerList(input)

    # Now convert that list into linked list
    dummyRoot = ListNode(0)
    ptr = dummyRoot  # 3.复制一个实例对象(可变变量,ptr属性变,dummyRoot 也会变,ptr赋值变,dummyRoot 不变)
    for number in numbers:  # [1,2,3]
        ptr.next = ListNode(number)  # 4. ListNode(val =0 ,next=ListNode(val =1 ,next=None))
        ptr = ptr.next  # 5. ptr=ListNode(val=1 ,next=None)
        # 第一层: dummyRoot: ListNode(val =0 ,next=ListNode(val =1 ,next=None))
        # 第二层: dummyRoot: ListNode(val =0 ,next=ListNode(val =1 ,next=ListNode(val =2 ,next=None)))

    ptr = dummyRoot.next  # 最后 ptr在dummyRoot的基础上去掉了val=0
    return ptr


def prettyPrintLinkedList(node):
    while node and node.next:
        print(str(node.val) + "->", end='')
        node = node.next

    if node:
        print(node.val)
    else:
        print("Empty LinkedList")


def main():
    def readlines():
        for line in sys.stdin:
            yield line.strip('\n')

    lines = readlines()  # 1.获取一个生成器
    while True:
        try:
            line = next(lines)  # 2.暂停代码, 通过next()获取生成输入的字符串列表
            node = stringToListNode(line)
            prettyPrintLinkedList(node)
        except StopIteration:
            break


if __name__ == '__main__':
    main()

pt结构如下图