2020/2/7

题目描述

• 给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。

示例

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后，链表变为 1->2->3->5.

相关标签

链表
双指针

解题思路

我们可以使用两个指针而不是一个指针。第一个指针从列表的开头向前移动 n+1 步，而第二个指针将从列表的开头出发。现在，这两个指针被 n 个结点分开。我们通过同时移动两个指针向前来保持这个恒定的间隔，直到第一个指针到达最后一个结点。此时第二个指针将指向从最后一个结点数起的第 n 个结点。我们重新链接第二个指针所引用的结点的 next 指针指向该结点的下下个结点。

• 时间复杂度：O(L)，该算法对含有 LL 个结点的列表进行了一次遍历。因此时间复杂度为 O(L)。
• 空间复杂度：O(1)，我们只用了常量级的额外空间。

源代码

// Definition for singly-linked list.
// #[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
//     }
//   }
// }
impl Solution {
    pub fn remove_nth_from_end(head: Option<Box<ListNode>>, n: i32) -> Option<Box<ListNode>> {
        let mut head_ans: Option<Box<ListNode>> = None;
        let mut p1 = &head;
        let mut p2 = &head;
        let mut p_head_ans = &mut head_ans;
        for _ in 0..n{
            if let Some(node_p2) = p2{
                p2 = &node_p2.next;
            }
        }
        while let Some(_) = p2{
            if let Some(node_p1) = p1{
                *p_head_ans = Some(Box::new(
                                        ListNode::new(
                                            node_p1.val)));
                if let Some(node_ans) = p_head_ans{
                    p_head_ans = &mut node_ans.next;
                }
            }
            if let Some(node_p1) = p1{
                p1 = &node_p1.next;
            }
            if let Some(node_p2) = p2{
                p2 = &node_p2.next;
            }
        }
        if let Some(node_p1) = p1{
            p2 = &node_p1.next;
        }
        while let Some(node_p2) = p2{
            *p_head_ans = Some(Box::new(
                                    ListNode::new(
                                        node_p2.val)));
            if let Some(node_ans) = p_head_ans{
                p_head_ans = &mut node_ans.next;
            }
            
            if let Some(node_p2) = p2{
                p2 = &node_p2.next;
            }
        }
        head_ans
    }
}

• 执行用时 :
  0 ms, 在所有 Rust 提交中击败了100.00%的用户
• 内存消耗 :
  1.9 MB, 在所有 Rust 提交中击败了100.00%的用户

总结

此题乃双指针的一次妙用，可以考虑采用闭包来减少代码的重复。