Java笔试(01) - 链表

链表

题目一:删除链表的倒数第N个节点

题解:在对链表进行操作时,一种常用的技巧是添加一个哑节点(dummy node),它的next 指针指向链表的头节点。这样一来,我们就不需要对头节点进行特殊的判断了。

例如,在本题中,如果我们要删除节点 yy,我们需要知道节点 yy 的前驱节点 xx,并将 xx 的指针指向 yy 的后继节点。但由于头节点不存在前驱节点,因此我们需要在删除头节点时进行特殊判断。但如果我们添加了哑节点,那么头节点的前驱节点就是哑节点本身,此时我们就只需要考虑通用的情况即可。

方法一:计算链表长度

思路与算法

一种容易想到的方法是,首先从头节点开始对链表进行一次遍历得到链表的长度L。随后再从头节点开始对链表进行一次遍历,当遍历到第L-n+1个节点时,它就是我们需要删除的节点。

为了与题目中的n保持一致,节点的编号从1开始,头节点为编号1的节点。

为了方便删除操作,可以从哑节点开始遍历L-n+1个节点。当遍历到第L-n+1个节点时,它的下一个节点就是需要删除的节点,这样只需要修改一次指针,就能完成删除操作。

代码如下:

class Solution {
    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;
    }
}

 方法二:栈

代码如下:

class Solution {
    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;
        ListNode ans = dummy.next;
        return ans;
    }
}

方法三:双指针

使用快慢指针。快指针比慢指针先n步,快慢指针同步前进,快指针碰到null时,慢指针就是待删除结点,需要维护一个pre结点,记录slow指针前面的结点,最后逻辑删除slow结点,返回head。

代码如下:

public class Solution {
    public ListNode removeNthFromEnd (ListNode head, int n) {
        // write code here
        ListNode fast = head;
        ListNode slow = head;
        ListNode pre = null;
        for (int i = 0; i < n; i++) {
            fast = fast.next;
        }
        while (fast != null) {
            fast = fast.next;
            pre = slow;
            slow = slow.next;
        }
        if (pre == null) {
            return slow.next;
        }
        pre.next = slow.next;
        return head;
    }
}

猜你喜欢

转载自blog.csdn.net/jianghao233/article/details/127764922