golang 优先队列构建和解题,(增加链表和树的打印)

链表的构造和打印

面试需要自己构造

• 链表

// 链表结构
type ListNode struct {
	Val  int
	Next *ListNode
}

//创建链表
func Create(nums []int) *ListNode {
	dummy := new(ListNode)
	cur := dummy
	for i := 0; i < len(nums); i++ {
		cur.Next = &ListNode{
			Val: nums[i],
		}
		cur = cur.Next
	}
	return dummy.Next
}

//打印链表
func Print(root *ListNode) {
	cur := root
	for cur != nil {
		fmt.Printf("%d->", cur.Val)
		cur = cur.Next
	}
}

树结构的构建和打印

type TreeNode struct {
	val         int
	left, right *TreeNode
}


// 中序方式构建树结构
func CreateTree(nums []int) *TreeNode {
	root := new(TreeNode)
	root.val = nums[0]
	queue := []*TreeNode{root}
	i := 1
	for i < len(nums) && len(queue) > 0 {
		node := queue[0]
		queue = queue[1:]

		// 左边
		if i < len(nums) {
			node.left = &TreeNode{val: nums[i]}
			queue = append(queue, node.left)
		}
		
		//右边
		i++
		if i < len(nums) {
			node.right = &TreeNode{val: nums[i]}
			queue = append(queue, node.right)
		}
		i++
	}
	return root
}


// 中序打印
func MidPrint(root *TreeNode) []int {
	var ret []int
	queue := []*TreeNode{root}
	for len(queue) != 0 {
		node := queue[0]
		queue = queue[1:]
		ret = append(ret, node.val)

		if node.left != nil {
			queue = append(queue, node.left)
		}
		if node.right != nil {
			queue = append(queue, node.right)
		}
	}
	return ret
}

func main() {
	nums := []int{1, 2, 3, 4, 5, 6}
	root := CreateTree(nums)
	ret := MidPrint(root)
	fmt.Println(ret)
}

前置条件

golang 优先队列几种构建方式,使用比起其他语言稍微有些复杂

• 简单构建方式

// 最小堆  s[0] 为堆顶  heap.Fix(&h,0) 可以从下标位置重新调整堆
type small []int 

func (s small)Less(i,j int)bool{
    return s[i]<s[j]
}

func (s small)Swap(i,j int){
    s[i],s[j] = s[j],s[i]
}

func (s small)Len()int{
    return len(s)
}

func (s *small)Push(x interface{}){
    tmp:=x.(int)
    *s = append(*s,tmp)
}

// 使用时 s:=new(small)

看到另外一种构建方式

func main(){
	nums:=[]int{1,6,4,8,2,7,2,4}
	h:=hp{}
	for _,v:=range nums{
		heap.Push(&h,v)
	}

	for h.Len()!=0{
		fmt.Printf("%d ",heap.Pop(&h))
	}
      // 输出为 8 7 6 4 4 2 2 1 
}

type hp struct{
	sort.IntSlice //底层是[]int{} 并且已经实现了 Len  Less默认升序  Swap  所以作为优先队列只需要实现 Push和Pop即可
       //type IntSlice []int

 }
 
 
//需要改降序则实现Less
func (h hp)Less(i ,j int)bool{
	return h.IntSlice[i]>h.IntSlice[j]
}

func (h *hp) Push(x interface{}) { h.IntSlice = append(h.IntSlice, x.(int)) }
func (h *hp) Pop() interface{} {
	old := h.IntSlice
	n := len(old)
	x := old[n-1]
	h.IntSlice = old[0 : n-1]
	return x
}

6178. 将区间分为最少组数

• 简单构建优先队列

func minGroups(intervals [][]int) int {
    if len(intervals)==0{
        return 0
    }
    
    //按照前面一个位置进行排序,相等按照后面的升序
    sort.Slice(intervals,func(i,j int)bool{
        if intervals[i][0]!=intervals[j][0]{
            return intervals[i][0]<intervals[j][0]
        }
        return intervals[i][1]>intervals[j][1]
    })
    
    //注意使用时用Heap.Push(&h,val)  Heap,Pop(&h)  关于绑定interface问题参考eggo
    s:=new(small)
    for i:=0;i<len(intervals);i++{
        if s.Len()==0||intervals[i][0]<=(*s)[0]{   //当前小于等于堆顶,则需要重新开一个组
            heap.Push(s,intervals[i][1])
        }else{
            (*s)[0] = intervals[i][1]   //可以放置,之前替换之前组的末尾数字,重新再平衡,或者 Heap.Pop() 后 Heap.Push()
            heap.Fix(s,0)
        }
    }
    return s.Len()
}

// 最小堆  s[0] 为堆顶  heap.Fix(&h,0) 可以从下标位置重新调整堆
type small []int 

func (s small)Less(i,j int)bool{
    return s[i]<s[j]
}

func (s small)Swap(i,j int){
    s[i],s[j] = s[j],s[i]
}

func (s small)Len()int{
    return len(s)
}

func (s *small)Push(x interface{}){
    tmp:=x.(int)
    *s = append(*s,tmp)
}

func (s *small)Pop()interface{}{
    x:=(*s)[len(*s)-1]
    (*s) = (*s)[:len(*s)-1]
    return x
}