aboutsummaryrefslogtreecommitdiff
path: root/leetcode.py
diff options
context:
space:
mode:
Diffstat (limited to 'leetcode.py')
-rw-r--r--leetcode.py19
1 files changed, 18 insertions, 1 deletions
diff --git a/leetcode.py b/leetcode.py
index 6fdbe28..29f7144 100644
--- a/leetcode.py
+++ b/leetcode.py
@@ -9,7 +9,7 @@ class TreeNode:
def treenode_build(vals):
- if not vals or vals[0] is None:
+ if not vals:
return None
root = TreeNode(vals[0])
q, i = deque([root]), 1
@@ -24,3 +24,20 @@ def treenode_build(vals):
q.append(curr.right)
i += 1
return root
+
+
+class ListNode:
+ def __init__(self, val=0, next=None):
+ self.val = val
+ self.next = next
+
+
+def listnode_build(vals):
+ if not vals or vals[0] is None:
+ return None
+ head = ListNode(vals[0])
+ curr = head
+ for i in vals[1:]:
+ curr.next = ListNode(i)
+ curr = curr.next
+ return head