代码随想录算法训练营第十三天|239. 滑动窗口最大值、347.前 K 个高频元素

239. 滑动窗口最大值

https://programmercarl.com/0239.%E6%BB%91%E5%8A%A8%E7%AA%97%E5%8F%A3%E6%9C%80%E5%A4%A7%E5%80%BC.html

思路 看了一个多小时大概搞懂了

func maxSlidingWindow(nums []int, k int) []int {
	ans:=make([]int, len(nums)-k+1)
	q:=make([]int,0)
	var push  func (x int)  
	push = func (x int)  {
		for len(q)>0&&q[len(q)-1]<x{
			q = q[:len(q)-1]
			
		}
		q = append(q,x)
	}
	for i := 0; i < k; i++ {
		push(nums[i])
		
	}

	ans[0]=q[0]
	for i := 1; i < len(nums)-k+1; i++ {
		//判定走不走 如果nums[i-1]==q[0]说明到了最大值 前面的可以舍去
		if nums[i-1]==q[0]{
			q= q[1:]
		}
		push(nums[i+k-1])
		ans[i]=q[0]
		
	}
	return ans


}

347.前 K 个高频元素

题目链接/文章讲解/视频讲解：https://programmercarl.com/0347.%E5%89%8DK%E4%B8%AA%E9%AB%98%E9%A2%91%E5%85%83%E7%B4%A0.html

思路： 搞不定 慢慢来 标记一下📌

func topKFrequent(nums []int, k int) []int {
    map_num:=map[int]int{}
    //记录每个元素出现的次数
    for _,item:=range nums{
        map_num[item]++
    }
    h:=&IHeap{}
    heap.Init(h)
    //所有元素入堆，堆的长度为k
    for key,value:=range map_num{
        heap.Push(h,[2]int{key,value})
        if h.Len()>k{
            heap.Pop(h)
        }
    }
    res:=make([]int,k)
    //按顺序返回堆中的元素
    for i:=0;i<k;i++{
        res[k-i-1]=heap.Pop(h).([2]int)[0]
    }
    return res
}

//构建小顶堆
type IHeap [][2]int

func (h IHeap) Len()int {
    return len(h)
}

func (h IHeap) Less (i,j int) bool {
    return h[i][1]<h[j][1]
}

func (h IHeap) Swap(i,j int) {
    h[i],h[j]=h[j],h[i]
}

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