python定义链表结构 将数组转为链表

import sys 
# 定义链表的数据结构
class ListNode:
    def __init__(self, x):
        self.val = x 
        self.next = None 
# 将数组转为链表结构
def generate_node(nums):
    # 定义一个哑节点
    p_head = ListNode(-1)
    temp_head = p_head 
    for num in nums:
        temp = ListNode(num) 
        temp_head.next = temp 
        temp_head = temp_head.next 
    return p_head.next 
if __name__ == "__main__":
    nums_1 = [4, 1, 8, 4, 5]
    # nums_2 = [5, 0, 1, 8, 4, 5]
    headA = generate_node(nums_1)
    # headB = generate_node(nums_2)
    while headA!=None:
        print(headA.val)
        headA = headA.next