面试必考真题-算法篇 牛客网
链表 双指针
题目描述
给定一个链表,删除链表的倒数第n个节点并返回链表的头指针
例如,
给出的链表为:1->2->3->4->5, n= 2.
删除了链表的倒数第n个节点之后,链表变为1->2->3->5.
备注:
题目保证n一定是有效的
请给出请给出时间复杂度为\ O(n) O(n)的算法。
题目分析
按照之前寻找倒数第n个节点的思路进行查找。
找的同时注意记录倒数第n个节点的前一个节点,找到后将其前一个节点指向其下一个节点即可。
返回头节点的时候注意删除的节点是否是头节点,如果是头节点需要判断其头节点的下一个节点是否存在,存在则下一个节点作为头节点返回,不存在则直接返回null。
这种头节点不存在下一个节点的情况是链表长度为1的情况。
下面是Java代码
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* }
*/
public class Solution {
/**
*
* @param head ListNode类
* @param n int整型
* @return ListNode类
*/
public ListNode removeNthFromEnd (ListNode head, int n) {
// write code here
if(head == null){
return null;
}
ListNode tmphead = head;
ListNode fast = head;
ListNode slow = head;
ListNode pre = slow;
while(fast!=null && n>0){
fast = fast.next;
--n;
}
if(n!=0){
return null;
}
while(fast!=null){
if(fast.next == null){
pre = slow;
}
fast = fast.next;
slow = slow.next;
}
//slow节点即为找到的倒数第n个节点。
//将找到slow节点的前一个节点指向slow的下一个节点,即删除了slow节点。
pre.next = slow.next;
//如果找到的节点是头节点,则判断下头节点是否存在下一个节点,存在则作为新的头节点返回,不存在则返回null。
if(slow == tmphead){
if(tmphead.next == null){
return null;
}else{
tmphead = tmphead.next;
}
}
return tmphead;
}
}