util.py 762 B

12345678910111213141516171819202122232425262728293031323334353637
  1. class ListNode(object):
  2. def __init__(self, val=0, _next=None, pre=None):
  3. self.val = val
  4. self.next = _next
  5. self.pre = pre
  6. def __str__(self):
  7. res_str = ''
  8. tmp = self
  9. while tmp:
  10. res_str += str(tmp.val)
  11. if tmp.next:
  12. res_str += ","
  13. tmp = tmp.next
  14. return res_str
  15. def get_node(arr: list):
  16. dummy = ListNode(0)
  17. curr = dummy
  18. for v in arr:
  19. curr.next = ListNode(v)
  20. curr = curr.next
  21. return dummy.next
  22. def get_arr(node: ListNode):
  23. res_arr = []
  24. while node:
  25. res_arr.append(node.val)
  26. node = node.next
  27. return res_arr
  28. if __name__ == '__main__':
  29. head = get_node([1, 2, 3, 4, 5])
  30. print(head)