题目难度:
Medium
题目要求:
给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
示例 1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
算法思路:
方法一:计算链表长度
1.从头节点开始,如果非空,长度+1,直至计算出链表的长度length;
2.利用循环,将指针定位到第length-n+1的节点,为了与倒数第n个节点匹配,循环应从i=1开始;
3.然后用cur.next=cur.next.next,将倒数第n个节点删掉,最后返回头节点即可。
代码:
//基本结构:
// class ListNode { //类名 :Java类就是一种自定义的数据结构
// int val; //数据 :节点数据
// ListNode next; //对象 :引用下一个节点对象。在Java中没有指针的概念,Java中的引用和C语言的指针类似
// }
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode first = new ListNode(0, head);
int Length = getLength(head);
ListNode cur = first;
for (int i = 1; i < Length - n + 1; ++i) {
cur = cur.next;
}
cur.next = cur.next.next;
ListNode ans = first.next;
return ans;
}
public int getLength(ListNode head) {
int length = 0;
while (head != null) {
++length;
head = head.next;
}
return length;
}
}
时间复杂度:O(N)
空间复杂度:O(1)
方法二:栈
遍历链表时,将节点依次入栈,然后依次出栈,直到倒数第n个节点,然后将该节点删除后,返回头节点即可。
代码:
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode first = new ListNode(0, head);
Deque<ListNode> stack = new LinkedList<ListNode>();
ListNode cur = first;
while (cur != null) {
stack.push(cur);
cur = cur.next;
}
for (int i = 0; i < n; ++i) {
stack.pop();
}
ListNode prev = stack.peek();//返回栈顶元素
prev.next = prev.next.next;
ListNode ans = first.next;
return ans;
}
}
时间复杂度:O(N)
空间复杂度:O(N)