刚学 C 语言的时候,学到指针这一章,就会有让我们写链表的需求,头插法,尾插法,翻转链表,合并链表,约瑟夫环等等。
学的不亦乐乎,但是 对于指针刚学的时候,真是摸不着脑壳,不知道 xdm 会有这种感受吗,动不动就段错误。
今天,我们来看看 golang 写链表是有多爽。
思路大概是这样的:
- 这里我们简单写,就不写循环链表了
- 需要一个链表结构体,这个结构体得有头指针,有尾指针
- 需要一个节点的结构体,这个里面有节点的数据,节点的下一个指针
- 我们这就来实现 头插法 和 尾插法吧
定义数据结构
- 定义链表结构
type MyList struct {
head *ListNode
tail *ListNode
}
- 定义节点结构
type ListNode struct {
num int
next * ListNode
}
实现头插法
所谓链表,就像排队一样,队员和队员之间用线连起来,但是这个线是有一个明确方向的,头插法,就是从链表头插入元素,新元素成为头
func (list *MyList) HeadInsert(num int) {
// 初始化一个节点,填入数据,和下一个指针指向 空
node := &ListNode{
num: num,
next: nil,
}
// 判断尾巴 和 头都是空,那么说明这一次是第 一个节点,头和尾都指向这个节点即可
if list.tail == nil && list.head == nil {
list.tail = node
list.head = node
return
}
// 头插,从头插入
node.next = list.head
list.head = node
}
实现尾插法
尾插法就是从链表尾巴插入元素,新元素成为尾巴
func (list *MyList) TailInsert(num int) {
// 初始化一个节点,填入数据,和下一个指针指向 空
node := &ListNode{
num: num,
next: nil,
}
// 判断尾巴 和 头都是空,那么说明这一次是第 一个节点,头和尾都指向这个节点即可
if list.tail == nil && list.head == nil {
list.tail = node
list.head = node
return
}
// 尾插,从尾插入
list.tail.next = node
list.tail = node
}
看到上述写的尾插法和头插法是不是写法很类似,其实就是后面的几行代码逻辑不一样而已,原理都是一样的
go 里面无论是使用指针,还是使用结构体,都是使用 .
,而不像 C 里面使用 ->
遍历链表
遍历链表就非常简单了,一个一个打出来就可以了
func (list *MyList) PrintList() {
if list == nil{
fmt.Println("list is nil")
return
}
tmp := list.head
for tmp != nil{
fmt.Println(tmp.num)
tmp = tmp.next
}
}
联合上述方法一起跑一遍
我们联合上述的代码,拼拼凑凑来实现我们的链表
- (头插或者尾插)插入 10 个数据
- 遍历一下链表
func main() {
// 插入 10 个数
l := MyList{}
for i:=0;i