Swapping Nodes in a Linked List

lcpy gen -n 1721   # by problem number
lcpy gen -s swapping_nodes_in_a_linked_list   # by problem name

Input: head = [1,2,3,4,5], k = 2
Output: [1,4,3,2,5]

Input: head = [7,9,6,6,7,8,3,0,9,5], k = 5
Output: [7,9,6,6,8,7,3,0,9,5]

from leetcode_py import ListNode


class Solution:
    # Time: O(n)
    # Space: O(1)
    def swap_nodes(self, head: ListNode[int] | None, k: int) -> ListNode[int] | None:
        if head is None:
            return head

        n = 1
        node = head
        while node.next is not None:
            node = node.next
            n += 1

        first = head
        for _ in range(k - 1):
            if first.next is None:
                break
            first = first.next

        second = head
        for _ in range(n - k):
            if second.next is None:
                break
            second = second.next

        first.val, second.val = second.val, first.val
        return head