solve 21

1 files changed, 1 insertions, 40 deletions

diff --git a/src/bin/p0002.rs b/src/bin/p0002.rs
index ac028a2..d8acf94 100644
--- a/src/bin/p0002.rs
+++ b/src/bin/p0002.rs
@@ -1,18 +1,4 @@
-#[derive(PartialEq, Eq, Clone, Debug)]
-pub struct ListNode {
-  pub val: i32,
-  pub next: Option<Box<ListNode>>
-}
-
-impl ListNode {
-  #[inline]
-  fn new(val: i32) -> Self {
-    ListNode {
-      next: None,
-      val
-    }
-  }
-}
+use leetcode::list::*;
 
 struct Solution {}
 
@@ -54,31 +40,6 @@ impl Solution {
     }
 }
 
-fn vector2list(vec: Vec<i32>) -> Option<Box<ListNode>> {
-    let mut result: Option<Box<ListNode>> = None;
-    for i in 0..vec.len() {
-        let mut new_result = Some(Box::new(ListNode::new(vec[vec.len() - i - 1])));
-        new_result.as_mut().unwrap().next = result;
-        result = new_result;
-    }
-    result
-}
-
-fn list2vector(lst: Option<Box<ListNode>>) -> Vec<i32> {
-    let mut result: Vec<i32> = Vec::new();
-    let mut cur = &lst;
-    loop {
-        match cur {
-            None => break,
-            Some(node) => {
-                result.push(node.val);
-                cur = &node.next;
-            },
-        }
-    }
-    result
-}
-
 fn test(l1: Vec<i32>, l2: Vec<i32>) -> Vec<i32> {
     list2vector(Solution::add_two_numbers(vector2list(l1), vector2list(l2)))
 }