给你一个链表,删除链表的倒数第 n
个结点,并且返回链表的头结点。
示例 1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
提示:
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
解题思路:
1、使用双指针发找到倒数n个节点。双指针分别记为F(fast)指针,S(slow)指针。最初S和F指针全部指向头节点
2、开始时候F指针先走n步,S指针原地不动
3、接着S,F指针全部往前走。直到F指针达到末尾位置的下一个节点,此时的S指针指向的是需要删除的节点
4、让被删节点的前一个节点的next指向S的next
Java代码
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode fast = head;
ListNode slow = head;
for(int i =0; i<n; i++){
fast = fast.next;
}
ListNode prev = null;
while(fast!=null){
fast = fast.next;
prev = slow;
slow = slow.next;
}
if(prev!=null){
prev.next = slow.next;
return head;
}else{
if(head.next==null){
return null;
}else{
return head.next;
}
}
}
}
Python代码
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def removeNthFromEnd(self, head, n):
"""
:type head: ListNode
:type n: int
:rtype: ListNode
"""
fast = head
slow = head
for i in range(n):
fast = fast.next
prev = None
while fast is not None:
fast = fast.next
prev = slow
slow = slow.next
if prev is not None:
prev.next = slow.next
return head
else:
if head.next is None:
return None
else:
return head.next
总结:
1 F指针在这里起到了指引的作用,是为了告诉S指针有没有达到末尾位置
2 方法只遍历一遍链表