題目描述:
給你一個(gè)鏈表,刪除鏈表的倒數(shù)第 n 個(gè)結(jié)點(diǎn),并且返回鏈表的頭結(jié)點(diǎn)。
進(jìn)階:你能嘗試使用一趟掃描實(shí)現(xiàn)嗎?
一、基礎(chǔ)類
public static class ListNode {
int val;
ListNode next;
ListNode() {
}
ListNode(int val) {
this.val = val;
}
ListNode(int val, ListNode next) {
this.val = val;
this.next = next;
}
}
二、個(gè)人拙見
public ListNode removeNthFromEnd(ListNode head, int n) {
// 自建一個(gè)頭節(jié)點(diǎn),應(yīng)對(duì)些特殊情況,dummy好好笑的起名
ListNode right = head, dummy = new ListNode(0, head);
ListNode left = dummy;
int gap = 0;
while (right.next != null){
++gap;
right = right.next;
if(gap >= n){
left = left.next;
}
}
ListNode del = left.next;
left.next = left.next.next;
del = null;
return dummy.next;
}
直接一步到位,直接完成了進(jìn)階版:用雙指針,一次遍歷完成解題。
三、官方解答
方法一、計(jì)算鏈表長(zhǎng)度
思路與算法
一種容易想到的方法是,我們首先從頭節(jié)點(diǎn)開始對(duì)鏈表進(jìn)行一次遍歷,得到鏈表的長(zhǎng)度 L。隨后我們?cè)購念^節(jié)點(diǎn)開始對(duì)鏈表進(jìn)行一次遍歷,當(dāng)遍歷到第 L?n+1 個(gè)節(jié)點(diǎn)時(shí),它就是我們需要?jiǎng)h除的節(jié)點(diǎn)。
代碼
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(0, head);
int length = getLength(head);
ListNode cur = dummy;
for (int i = 1; i < length - n + 1; ++i) {
cur = cur.next;
}
cur.next = cur.next.next;
ListNode ans = dummy.next;
return ans;
}
public int getLength(ListNode head) {
int length = 0;
while (head != null) {
++length;
head = head.next;
}
return length;
}
復(fù)雜度分析
- 時(shí)間復(fù)雜度:O(L),其中 L 是鏈表的長(zhǎng)度。
- 空間復(fù)雜度:O(1)。
方法二、棧
思路與算法
我們也可以在遍歷鏈表的同時(shí)將所有節(jié)點(diǎn)依次入棧。根據(jù)棧「先進(jìn)后出」的原則,我們彈出棧的第 n 個(gè)節(jié)點(diǎn)就是需要?jiǎng)h除的節(jié)點(diǎn),并且目前棧頂?shù)墓?jié)點(diǎn)就是待刪除節(jié)點(diǎn)的前驅(qū)節(jié)點(diǎn)。這樣一來,刪除操作就變得十分方便了。
代碼
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(0, head);
Deque<ListNode> stack = new LinkedList<ListNode>();
ListNode cur = dummy;
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;
return dummy.next;
}
復(fù)雜度分析
- 時(shí)間復(fù)雜度:O(L),其中 L 是鏈表的長(zhǎng)度。
- 空間復(fù)雜度:O(L),其中 L 是鏈表的長(zhǎng)度。主要為棧的開銷。
方法三、雙指針
思路與算法
由于我們需要找到倒數(shù)第 n 個(gè)節(jié)點(diǎn),因此我們可以使用兩個(gè)指針 first 和 second 同時(shí)對(duì)鏈表進(jìn)行遍歷,并且 first 比 second 超前 n 個(gè)節(jié)點(diǎn)。當(dāng) first 遍歷到鏈表的末尾時(shí),second 就恰好處于倒數(shù)第 n 個(gè)節(jié)點(diǎn)。
代碼
class Solution {
public ListNode removeNthFromEnd(ListNode head, int n) {
ListNode dummy = new ListNode(0, head);
ListNode first = head;
ListNode second = dummy;
for (int i = 0; i < n; ++i) {
first = first.next;
}
while (first != null) {
first = first.next;
second = second.next;
}
second.next = second.next.next;
ListNode ans = dummy.next;
return ans;
}
}
復(fù)雜度分析
- 時(shí)間復(fù)雜度:O(L),其中 L 是鏈表的長(zhǎng)度。
- 空間復(fù)雜度:O(1)。