面试题22. 链表中倒数第k个节点

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

 示例:

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

初步解决方法:

通过两个指针,一个指针用于遍历链表长度,然后再用一个指针来找到输出链表的结点位置。

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if (head == NULL) return head;
        ListNode* p = head;
        ListNode* q = head;
        int sum = 0;
        while (p){
            sum++;
            p = p->next;
        }
        if (k > sum) return head;
        int index = sum - k;
        while (index--){
            q = q->next;
        }
        return q;
    }
};

进一步思考

快慢指针。

class Solution {
    public ListNode getKthFromEnd(ListNode head, int k) {
        ListNode slow = head;
        ListNode fast = head;
        while (fast != null && k-- > 0) {
            fast = fast.next;
        }
        while(fast != null) {
            slow = slow.next;
            fast = fast.next;
        }
        return slow;
    }
}

猜你喜欢

转载自www.cnblogs.com/shawjoe/p/12318118.html
今日推荐