golang怎么实现反转链表

PHPz

发布时间：2023-04-23 10:22:20

1244人浏览过

来源于php中文网

原创

反转链表是常见的一个问题，在程序面试中也经常被提到。它是一道经典的算法问题，应用广泛，可以用于快速反转链表的顺序。本文将介绍使用golang语言实现反转链表的算法和步骤。

定义单链表节点

在开始实现反转链表之前，我们需要先定义一个单链表的节点。一个节点包含两个非常重要的部分：数据域和指针域。数据域用来存储节点的值，指针域用来指向下一个节点。

在golang中，我们可以使用struct结构体来定义一个单链表节点。结构体中包含两个属性：Val，用来表示当前节点的值，Next，用来表示指向下一个节点的指针。

type ListNode struct {

Val  int
Next *ListNode

}

立即学习“go语言免费学习笔记（深入）”；

绘蛙AI修图

绘蛙平台AI修图工具，支持手脚修复、商品重绘、AI扩图、AI换色

下载

单链表反转

现在我们已经定义了单链表的节点，下一步是实现反转链表的算法。反转链表的关键是遍历链表并更改每个节点的指针指向。

我们可以从头开始遍历链表中的每个节点，并且依次改变它们的“Next”指针，指向前一个节点。这样就可以实现链表的反转了。

反转链表的算法步骤如下：

（1）定义两个指针：pre和cur，分别指向第一个节点和第二个节点。pre为前一个节点，cur为当前节点。

（2）遍历链表，分别将当前节点的Next指针指向前一个节点pre。

（3）向后移动指针，将pre指向当前节点，cur指向下一个节点。

（4）重复步骤2和3，直到遍历完整个链表。

实现代码如下：

func reverseLinkedList(head ListNode) ListNode {

var pre *ListNode
cur := head
for cur != nil {
    next := cur.Next
    cur.Next = pre
    pre = cur
    cur = next
}
return pre

}

立即学习“go语言免费学习笔记（深入）”；

反转链表的测试代码

为了验证反转链表的正确性，我们编写一些测试代码来执行。

func TestReverseLinkedList(t *testing.T) {

head := &ListNode{Val: 1}
node1 := &ListNode{Val: 2}
node2 := &ListNode{Val: 3}
node3 := &ListNode{Val: 4}
node4 := &ListNode{Val: 5}

head.Next = node1
node1.Next = node2
node2.Next = node3
node3.Next = node4

newHead := reverseLinkedList(head)

assert.Equal(t, newHead.Val, 5)
assert.Equal(t, newHead.Next.Val, 4)
assert.Equal(t, newHead.Next.Next.Val, 3)
assert.Equal(t, newHead.Next.Next.Next.Val, 2)
assert.Equal(t, newHead.Next.Next.Next.Next.Val, 1)

}

立即学习“go语言免费学习笔记（深入）”；

反转部分链表

除了反转整个链表之外，我们还可以反转链表中的一部分。例如，反转链表中第m个节点到第n个节点的部分。我们只需要在反转整个链表的基础上稍作修改即可。

我们可以首先遍历到第m-1个节点，pre指针指向该节点，cur指向第m个节点。然后，我们执行反转链表的步骤，直到反转到第n个节点为止。

实现代码如下：

func reverseBetween(head ListNode, m int, n int) ListNode {

dummy := &ListNode{0, head}
pre := dummy

for i := 1; i < m; i++ {
    pre = pre.Next
}

cur := pre.Next
for i := m; i < n; i++ {
    next := cur.Next
    cur.Next = next.Next
    next.Next = pre.Next
    pre.Next = next
}

return dummy.Next

}

立即学习“go语言免费学习笔记（深入）”；

反转部分链表的测试代码

为了验证反转部分链表的正确性，我们编写一些测试代码进行验证。

func TestReverseBetween(t *testing.T) {

head := &ListNode{Val: 1}
node1 := &ListNode{Val: 2}
node2 := &ListNode{Val: 3}
node3 := &ListNode{Val: 4}
node4 := &ListNode{Val: 5}

head.Next = node1
node1.Next = node2
node2.Next = node3
node3.Next = node4

newHead := reverseBetween(head, 2, 4)

assert.Equal(t, newHead.Val, 1)
assert.Equal(t, newHead.Next.Val, 4)
assert.Equal(t, newHead.Next.Next.Val, 3)
assert.Equal(t, newHead.Next.Next.Next.Val, 2)
assert.Equal(t, newHead.Next.Next.Next.Next.Val, 5)

}

立即学习“go语言免费学习笔记（深入）”；