Swift 删除链表的倒数第N个节点 - LeetCode

LeetCode

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

给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。
示例:

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

当删除了倒数第二个节点后，链表变为 1->2->3->5.

说明:
给定的 n 保证是有效的。

方案：

构建双指针first与sec，first先走n步，然后一同运动，当first指向表尾，sec指向的next即是倒数第N个节点，删除即可(next指向next的next，这里的删除相当于跳过)。

代码：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     public var val: Int
 *     public var next: ListNode?
 *     public init(_ val: Int) {
 *         self.val = val
 *         self.next = nil
 *     }
 * }
 */
class Solution {
  func removeNthFromEnd(_ head: ListNode?, _ n: Int) -> ListNode? {
    var first = head
    for _ in 0..<n {
        first = first?.next
    }
    if first == nil {
        return head?.next
    }
    var sec = head
    while first?.next != nil {
        first = first?.next
        sec = sec?.next
    }
    sec?.next = sec?.next?.next
    return head
  }
}