LeetCode笔记:19. Remove Nth Node From End of List

来源:互联网 发布:网站权限数据库设计 编辑:程序博客网 时间:2024/05/21 10:52

问题:

Given a linked list, remove the nth node from the end of list and return its head.

For example,

Given linked list: 1->2->3->4->5, and n = 2.

After removing the second node from the end, the linked list becomes 1->2->3->5.

Note:
Given n will always be valid.
Try to do this in one pass.

大意:

给出一个链表,移除链表的倒数第n个节点并返回链表头。

例子,

给出链表: 1->2->3->4->5, 以及 n = 2.
在移除倒数第二个节点后,列表变为了 1->2->3->5。

注意:
给出的n一定是有效的。
尝试在一轮循环中做。

思路:

题目的难点在于你不知道遍历到第几个节点时是要删除的倒数第n个节点。

我的做法很笨,遍历一遍记录所有节点的值和位置,然后重新根据值和位置创建新的链表,跳过要删除的那个位置的节点,因为此时知道总节点数了就可以推出是第几个节点了。在操作时要注意一些特殊情况,比如只有一个节点时、删除头结点时要怎么处理。

代码(Java):

/** * Definition for singly-linked list. * public class ListNode { *     int val; *     ListNode next; *     ListNode(int x) { val = x; } * } */public class Solution {    public ListNode removeNthFromEnd(ListNode head, int n) {        int[] nodeVal = new int[100];        int[] nodeIndex = new int[100];        nodeVal[0] = head.val;        nodeIndex[0] = 0;        int index = 1;        while (head.next != null) {            head = head.next;            nodeVal[index] = head.val;            nodeIndex[index] = index;            index++;        }        ListNode newHead;        int begin = 0;        if (index == 1) return null;        else if (index == n) {            newHead = new ListNode(nodeVal[1]);            begin = 1;        } else newHead = new ListNode(nodeVal[0]);        ListNode tempNode = newHead;        for (int i = begin+1; i < index; i++) {            if (i != index - n) {                ListNode newNode = new ListNode(nodeVal[i]);                tempNode.next = newNode;                tempNode = newNode;            }        }        return newHead;    }}

他山之石:

public ListNode removeNthFromEnd(ListNode head, int n) {    ListNode start = new ListNode(0);    ListNode slow = start, fast = start;    slow.next = head;    //Move fast in front so that the gap between slow and fast becomes n    for(int i=1; i<=n+1; i++)   {        fast = fast.next;    }    //Move fast to the end, maintaining the gap    while(fast != null) {        slow = slow.next;        fast = fast.next;    }    //Skip the desired node    slow.next = slow.next.next;    return start.next;}

看一下这个巧妙的做法,他设了快慢两个标记,初始都在头结点,快的先往后遍历,遍历到与头结点相差为n的时候停止,然后快的和慢的一起往后走,直到快的走到了链表尾节点打止,这时候快慢两个节点间相差的节点数正好是n,也就是说慢的所在的下一个节点正好是要删除的节点,直接跳过去就可以了,一遍遍历完成,很棒。


合集:https://github.com/Cloudox/LeetCode-Record
版权所有:http://blog.csdn.net/cloudox_

0 0
原创粉丝点击