首页 > 技术文章 > leetcode 练习--反转链表

cai-study-ai-cv 2020-11-29 10:23 原文

最近开始学习数据结构和算法的学习,也自然开始在 leetcode 上练习,所以每周大概会分享做过的leetcode 练习,尽量做到每天更新一道题目。

作为 leetcode 练习笔记的第一道题目,选择了一道很经典的题目,反转链表。这是 leetcode 上的 206 题,链接如下:

https://leetcode-cn.com/problems/reverse-linked-list/

示例

Input: 1->2->3->4->5->NULL
Output: 5->4->3->2->1->NULL

思路

反转一个单链表,首先肯定需要遍历这个单链表,在遍历的时候就希望修改当前结点的 next 指针,指向其前一个结点,因此肯定需要一个保存前一个结点的变量,也就是反转后链表的头部指针。

实现的思路应该是这样的:

  1. 首先定义一个 prev 保存前一个结点,curr 保存当前结点,然后还有一个 nxt 保存下一个结点,其中 prev 就是最终的反转链表的头结点;
  2. 先让 nxt 保存下一个结点;
  3. 然后改变 currnext 指针,指向前一个结点,即 prev ;
  4. 接着,让 prev = curr
  5. 最后,就是让 curr = nxt,指向下一个结点
  6. 重复 2-5 步,直到当前结点为空。

下图展示了上述几个步骤的过程:

实现

利用 python 的特性,实现的时候关键代码其实就一行即可。

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None


class Solution:
    def reverseList(self, head: ListNode) -> ListNode:
        pre, cur = None, head
        while cur:
            pre, pre.next, cur = cur, pre, cur.next
        return pre

github地址:

https://github.com/ccc013/DataStructe-Algorithms_Study/blob/master/Python/Leetcodes/linked_list/206_Reverse_Linked_List.py

推荐阅读