删除链表倒数第N个节点
1.问题
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
2.算法
暴力破解法:先计算得出链表的长度M,然后将链表长度和所给倒数位置相减,即loc=M-n-1,得出删除节点的前一个节点,然后就可以删除了。具体代码如下:
/**
* 链表节点的定义
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
public class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
if(head==null)
return null;
ListNode temp=head,pre=null;
int listlong=0;
while(temp!=null) { //求出链表长度
listlong++;
temp=temp.next;
}
//如果删除位置和链表长度一致,即可直接返回首节点的下一个节点
if(listlong==n)
return head.next;
temp=head; //重新回到首节点
for (int i = 0; i < listlong-n-1; i++) {
temp=temp.next;
}
pre=temp;
temp=temp.next;
pre.next=temp.next;
return head;
}
}
这样算法很容易得出时间复杂度是O(M),空间复杂度为O(1),只用了一个临时指针的空间。
但是这样的话是两次遍历,因此这个算法可以进行优化,变成只需要一次遍历即可。
优化算法:
设定双指针P,Q,当指针P指向链表节点最后指向的null时,它与指针Q相隔的元素刚好为n个,那么指针Q的下一个节点就是我们刚好要删除的节点。如下图所示(图源来自于leetcode第19题的程序员吴师兄的解释,链接:动图图解以及C++代码):
代码如下:
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode first = head;
ListNode second = head;
int i = 1;
for (; i < n; i++) { //移动p指针
if (first != null) {
first = first.next;
} else {
break;
}
}
if (i < n) { //如果i<n,说明n大于链表长度,不用删除
return head;
}
//刚好n使链表的长度,则直接删除头节点即可
if (first.next == null) {
head = head.next;
} else {
first = first.next; //使指针p和q之间相隔n个元素
while (first.next != null) {
first = first.next;
second = second.next;
}
second.next = second.next.next;
}
return head;
}
}
这样的时间复杂度也是O(M),空间复杂度也是O(1),但是它只需要一次遍历,因此对于长链表来说的话,这样的提升也是很大的。