链表-简单:链表中倒数第k个节点 By Swift

120 阅读1分钟
  • 题目

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

示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
  • 思路

双指针。fast 先走 k 步,然后 slow 和 fast 再一起走,fast 走到头, slow 指向的即为倒数第 k 个节点。

  • 代码
class Solution {
    func getKthFromEnd(_ head: ListNode?, _ k: Int) -> ListNode? {
        var slow = head, fast = head
        var step = k
        
        while step > 0 {
            fast = fast?.next
            step -= 1
        }
        
        while fast != nil {
            slow = slow?.next
            fast = fast?.next
        }
        
        return slow
    }
}