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%的用户
总结
此题乃双指针的一次妙用,可以考虑采用闭包来减少代码的重复。