《剑指offer》第15题 链表中倒数第K个结点(Java版答案)

题目:

输入一个链表,输出链表中倒数第K个结点。

答案:

常规思路就是要经过两次遍历,不够高效。

先遍历一遍链表,得到链表的长度n,再减去k,得到结点的正数位置,最后再遍历得到这个结点。

 

双指针法。只需要遍历一次链表

第一个指针p1先走k步

然后第二个指针p2与p1同时走,这样p1和p2之间一直保持k个结点的距离。

因此直到p1走到最后一个结点,此时p2正好位于倒数第k个结点。

public ListNode getLastK(ListNode head, int k) {
        if (head == null || k <= 0) {
            throw new RuntimeException("非法输入");
        }

        ListNode p1 = head;
        ListNode p2 = head;

        //p1先走k步
        for (int i = 0; i < k; i++) {
            if (p1.next != null) {
                p1 = p1.next;
            } else {
                return null;
            }

        }

        //p1和p2同时走,直到p1走到最后结点,此时p2刚好指向倒数第k个结点
        while (p1.next != null) {
            p1 = p1.next;
            p2 = p2.next;
        }

        return p2;
    }

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章