剑指 Offer 22. 链表中倒数第k个节点

2022-06-24  本文已影响0人  Sun东辉

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例:

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

返回链表 4->5.

解题思路:思考是否可以利用数据结构本身的特性

type ListNode struct {
    Val  int
    Next *ListNode
}

func getKthFromEnd(head *ListNode, k int) *ListNode {
    len := 0
    m := make(map[int]*ListNode)
    m[0] = head
    for head.Next != nil {
        len++
        m[len] = head.Next
        head = head.Next
    }
    return m[len-k+1]
}

解题思路二:快慢指针

快指针领先慢指针 k 个节点,当快指针为 nil 时,慢指针的状态即是我们想要的结果。

type ListNode struct {
    Val  int
    Next *ListNode
}

func getKthFromEnd(head *ListNode, k int) *ListNode {
    fast, slow := head, head
    for fast != nil && k > 0 {
        fast = fast.Next
        k--
    }
    for fast != nil {
        fast = fast.Next
        slow = slow.Next
    }
    return slow
}

来源:力扣(LeetCode) 链接:https://leetcode.cn/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof

上一篇 下一篇

猜你喜欢

热点阅读