206. 反转链表简单迭代递归

Posted 2023-02-20 pre_eminent

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

---

示例 1：

输入：head = [1,2,3,4,5]
输出：[5,4,3,2,1]

---

 

示例 2：

输入：head = [1,2]
输出：[2,1]

---

示例 3：

输入：head = []
输出：[]
 

---

提示：

链表中节点的数目范围是 [0, 5000]
-5000 <= Node.val <= 5000
 

进阶：链表可以选用迭代或递归方式完成反转。你能否用两种方法解决这道题？

---

思路

1. 迭代

2. 递归

---

解答：

/**
 * Definition for singly-linked list.
 * function ListNode(val, next) 
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * 
 */
/**
 * @param ListNode head
 * @return ListNode
 */
var reverseList = function(head) 
    return iterate(head);
;

// 方法2：递归 (反转 1 → 尾)
function recursion(node1) 
    // 递归终点
    if (!node1 || !node1.next) 
        return node1;
    

    // 先取出 node2
    const node2 = node1.next;
    // 假设 2 → 尾 已经反转 (不关心细节)
    const newHead = recursion(node2);
    // 反转 1 → 2
    node2.next = node1;
    node1.next = null;
    // 返回 新的头结点
    return newHead;


// 方法1：迭代
function iterate(node1) 
    // 健壮性判断
    if (!node1 || !node1.next) 
        return node1;
    

    // 先反转1 → 2
    let node2  = node1.next;
    node1.next = null;
    // 如果node2存在，继续反转
    while(node2) 
        // 同样方法反转2 → 3
        let node3 = node2.next;
        node2.next = node1;
        // 关键代码
        // 更新node1和node2
        node1 = node2;
        node2 = node3;
    
    return node1;