常见算法面试题:删除链表的倒数第N个节点

题目:给定一个链表,删除链表的倒数第 n 个节点并返回头结点。

说明:给的 n 始终是有效的。尝试一次遍历实现。

例子:
给定一个链表: 1->2->3->4->5, 并且 n = 2.
当删除了倒数第二个节点后链表变成了 1->2->3->5.

思路: 很简单的一道题,既然只允许遍历一次,且N一直有效,那么可以利用双指针解法(前指针、后指针),让前指针先走N步,再让两个在指针同时后移,直到前指针到达尾部,此时,后指针的下一个节点就是要被删除的节点了。

@Test
public void test() {
ListNode head = new ListNode(1);
ListNode listNode = removeNthFromEnd(head, 1);
System.out.println(listNode);

}

public ListNode removeNthFromEnd(ListNode head, int n) {

ListNode preNode = head;
ListNode curNode = head;

for (int i = 0; i < n; i++) {
curNode = curNode.next;
}

if (curNode == null) {
return preNode.next;
}

while (curNode.next != null) {
preNode = preNode.next;
curNode = curNode.next;
}

preNode.next = preNode.next.next;

return head;
}

class ListNode {
int val;
ListNode next;

ListNode(int x) {
val = x;
}
}

猜你喜欢

转载自blog.csdn.net/qq_31279347/article/details/82531870
今日推荐