力扣题目学习笔记(OC + Swift)206. 反转链表

Jarlen John2024-01-03 8:37

206. 反转链表

给你单链表的头节点 head ，请你反转链表，并返回反转后的链表。

示例

方法一、迭代

在遍历链表时，将当前节点的 next\textit{next}next 指针改为指向前一个节点。由于节点没有引用其前一个节点，因此必须事先存储其前一个节点。在更改引用之前，还需要存储后一个节点。最后返回新的头引用。

复杂度分析

时间复杂度：O(n)，其中 nnn 是链表的长度。需要遍历链表一次。

空间复杂度：O(1)。

Swift

swift 复制代码

//迭代法
    func reverseList(_ head: ListNode?) -> ListNode? {
        var previ: ListNode?
        var cur = head
        
        while cur != nil {
            let nex = cur?.next
            cur?.next = previ
            previ = cur
            cur = nex
        }
        
        return previ
    }

OC

c 复制代码

//迭代
-(ListNodeOC *)reverseList:(ListNodeOC *)head {
    ListNodeOC *pre = nil;
    ListNodeOC *cur = head;
    
    while (cur != nil) {
        ListNodeOC *nex = cur.next;
        cur.next = pre;
        pre = cur;
        cur = nex;
    }
    
    return pre;
}

方法二、递归

递归版本稍微复杂一些，其关键在于反向工作。假设链表的其余部分已经被反转，现在应该如何反转它前面的部分？

比较抽象，直观理解是把head和除head之外的部分看成两个部分，head已不需要翻转，另一部分需要调用自己继续完成翻转，最后重新组织前后依赖关系，注意新链表的尾部节点的next为空，否则会引入循环。

复杂度分析

时间复杂度：O(n)，其中 nnn 是链表的长度。需要遍历链表一次。

空间复杂度：O(n)。

Swift

swift 复制代码

    func reverseList(_ head: ListNode?) -> ListNode? {
        // 递归终止条件
        if head == nil || head?.next == nil {
            return head
        }
        
        let next = head?.next
        let node = reverseList(next)
        head?.next = nil
        next?.next = head
        return node
    }

OC

c 复制代码

-(ListNodeOC *)reverseList:(ListNodeOC *)head {
    //递归终止条件
    if (!head || !head.next) {
        return head;
    }

    ListNodeOC *next = head.next;
    ListNodeOC *node = [self reverseList:next];
    
    head.next = nil;
    next.next = head;
    
    return node;
}