diff --git a/2181_merge_nodes_in_between_zeros/Cargo.lock b/2181_merge_nodes_in_between_zeros/Cargo.lock new file mode 100644 index 0000000..2cbc0b0 --- /dev/null +++ b/2181_merge_nodes_in_between_zeros/Cargo.lock @@ -0,0 +1,7 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 4 + +[[package]] +name = "merge_nodes_in_between_zeros" +version = "0.1.0" diff --git a/2181_merge_nodes_in_between_zeros/Cargo.toml b/2181_merge_nodes_in_between_zeros/Cargo.toml new file mode 100644 index 0000000..0ea2c86 --- /dev/null +++ b/2181_merge_nodes_in_between_zeros/Cargo.toml @@ -0,0 +1,6 @@ +[package] +name = "merge_nodes_in_between_zeros" +version = "0.1.0" +edition = "2024" + +[dependencies] diff --git a/2181_merge_nodes_in_between_zeros/src/lib.rs b/2181_merge_nodes_in_between_zeros/src/lib.rs new file mode 100644 index 0000000..65afe52 --- /dev/null +++ b/2181_merge_nodes_in_between_zeros/src/lib.rs @@ -0,0 +1,39 @@ +#[derive(PartialEq, Eq, Clone, Debug)] +pub struct ListNode { + pub val: i32, + pub next: Option>, +} + +impl ListNode { + #[inline] + fn new(val: i32) -> Self { + ListNode { next: None, val } + } +} + +pub struct Solution; + +impl Solution { + pub fn merge_nodes(head: Option>) -> Option> { + let mut result_head = Some(Box::new(ListNode::new(0))); + let mut result_current_node = &mut result_head; + + let mut current_node = head.unwrap().next; + while let Some(node) = current_node.take() { + if node.val != 0 { + result_current_node.as_mut().unwrap().val += node.val; + } else { + if node.next.as_ref().is_none() { + break; + } + + result_current_node.as_mut().unwrap().next = Some(Box::new(ListNode::new(0))); + result_current_node = &mut result_current_node.as_mut().unwrap().next; + } + + current_node = node.next; + } + + result_head + } +}