Go数据结构--双向链表

package main

import "fmt"

//双向链表

// HeroNode 定义一个双向链表节点
type HeroNode struct {
	no int
	name string
	nickname string
	preNode *HeroNode
	nextNode *HeroNode
}

// InsertHeroNode 向链表添加节点
func InsertHeroNode(head *HeroNode,newHeroNode *HeroNode) {
	//1.先找到该节点的最后一个节点(即 next == nil的节点)
	//2.创建一个辅助节点[跑龙套，帮忙的]
	move := head
	for  {
		if move.nextNode == nil {
			//该节点为最后一个
			break
		}
		//不是最后一个，找下一个
		move = move.nextNode
	}
	//3.将newHeroNode加入到链表的最后
	move.nextNode = newHeroNode
	newHeroNode.preNode = move
}

// InsertHeroNode2 根据no编号从小到大插入
func InsertHeroNode2(head *HeroNode,newHeroNode *HeroNode) {
	move := head
	flag := true
	for  {
		if move.nextNode == nil {
			//没有下一个节点，也就是最后一个节点了
			break
			//}else if move.nextNode.no > newHeroNode.no {
		}else if move.nextNode.no >= newHeroNode.no {
			//即 是这种状态 move newHeroNode move.NextNode
			break
		}else if move.nextNode.no == newHeroNode.no{
			//已经有这个节点了
			flag = false
			break
		}
		move = move.nextNode
	}
	if !flag {
		fmt.Println("对不起，已存在no=",newHeroNode.no)
		return
	}else {
		newHeroNode.nextNode = move.nextNode //向下链接 move.nextNode
		newHeroNode.preNode = move //向上链接 move
		if move.nextNode != nil {
			move.nextNode.preNode = newHeroNode			//move不是最后一个
		}
		move.nextNode = newHeroNode
	}
}

// ListHeroNode 显示链表的所有节点信息
func ListHeroNode(head *HeroNode)  {
	//1.创建一个辅助节点[跑龙套，帮忙]
	move := head
	//空判断
	if move.nextNode == nil {
		fmt.Println("头节点next是空的，这是个空节点")
		return
	}
	//2.遍历这个链表
	for  {
		fmt.Printf("[%d,%s,%s]==>",move.nextNode.no,move.nextNode.name,move.nextNode.nickname)
		//继续查看下一个
		move = move.nextNode
		if move.nextNode == nil {
			break
		}
	}
}

// ListHeroNode2 逆序打印
func ListHeroNode2(head *HeroNode)  {
	//创建一个辅助节点[跑龙套，帮忙]
	move := head
	//空判断
	if move.nextNode == nil {
		fmt.Println("头节点next是空的，这是空节点")
		return
	}
	//1.让move定位到双向链表的结尾
	for  {
		if move.nextNode == nil {
			//move.nextNode就是最后一个
			break
		}
		move = move.nextNode //移动到最后一个
	}
	//2.遍历这个链表
	for  {
		fmt.Printf("[%d,%s,%s]==>",move.no,move.name,move.nickname)
		//查看上一个
		move = move.preNode
		if move.preNode == nil {
			break
		}
	}
}


// DelHeroNode 删除一个节点
func DelHeroNode(head *HeroNode,id int)  {
	//1.创建一个辅助节点[跑龙套，帮忙]
	move := head
	flag := false
	//找到要删除的节点的no，和move.nextNode的no比较
	for  {
		if move.nextNode == nil {
			//这个move是最后一个，停！,这说明最后都没找到这个id的节点
			break
		}else if move.nextNode.no == id {
			//找到啦
			flag = true
			break
		}
		move = move.nextNode//继续找下一个
	}

	if flag {
		//删除,剔除中间的move.nextNode
		move.nextNode = move.nextNode.nextNode
		if move.nextNode != nil {
			//确保move不是最后一个元素，如果是最后一个元素，move.nextNode没哟preNode
			move.nextNode.preNode = move
		}
	}else {
		fmt.Println("要删除id不存在")
	}
}


func main()  {
	//1.先创建一个头节点
	head := &HeroNode{}

	//2.创建一个新的HeroNode
	hero1 := &HeroNode{
		no: 1,
		name: "宋江",
		nickname: "及时雨",
	}
	hero2 := &HeroNode{
		no: 2,
		name: "卢俊义",
		nickname: "玉麒麟",
	}
	hero3 := &HeroNode{
		no: 3,
		name: "林冲",
		nickname: "豹子头",
	}
	hero4 := &HeroNode{
		no: 4,
		name: "吴用",
		nickname: "智多星",
	}

	//3.加入
	//InsertHeroNode(head,hero1)
	//InsertHeroNode(head,hero2)
	//InsertHeroNode(head,hero3)
	//InsertHeroNode(head,hero4)
	//
	//fmt.Println("顺序打印")
	//ListHeroNode(head)
	//fmt.Println()
	//
	//fmt.Println("逆序打印")
	//ListHeroNode2(head)
	//fmt.Println()

	//return
	InsertHeroNode2(head,hero3)
	InsertHeroNode2(head,hero1)
	InsertHeroNode2(head,hero4)
	InsertHeroNode2(head,hero2)

	fmt.Println("顺序打印")
	ListHeroNode(head)
	fmt.Println()

	fmt.Println("逆序打印")
	ListHeroNode2(head)
	fmt.Println()

	fmt.Println("删除")
	DelHeroNode(head,3)
	fmt.Println("顺序打印")
	ListHeroNode(head)
	fmt.Println()

	fmt.Println("逆序打印")
	ListHeroNode2(head)
	fmt.Println()
}