golang刷leetcode 前 K 个高频元素

func topKFrequent(nums []int, k int) []int {
   m:=make(map[int]int)
   for _,n:=range nums{
       m[n]++
   }
   h:=&heap{
       cap:k,
   }
   for k,v:=range m{
      h.push(d{num:k,count:v})
   }
   var v []int
   for _,i:=range h.data{
     v=append(v,i.num)
   }
   return v
}

type d struct{
        num int
        count int
}

type heap struct {
    data []d
    cap int
}

func (h*heap)push(da d){
   if len(h.data)>=h.cap{
       if da.count<=h.data[0].count{
           return
       }
       h.data[0]=da
       h.adjust(0)
       return
   }
    h.data=append(h.data,da)
   if len(h.data)==h.cap{
       for k:=h.cap/2;k>=0;k--{
           h.adjust(k)
       }
   }
}
func (h *heap)adjust(i int){
    for i<h.cap {
    l:=2*i+1
    r:=2*i+2
    if r<h.cap{
        if h.data[l].count<h.data[r].count &&  h.data[l].count< h.data[i].count{
            h.data[i],h.data[l]= h.data[l],h.data[i]
            i=l
        }else if  h.data[r].count< h.data[i].count{
             h.data[i],h.data[r]= h.data[r],h.data[i]
             i=r
        }else{
            break
        }
    }else if l< h.cap &&  h.data[l].count< h.data[i].count{
            h.data[i],h.data[l]= h.data[l],h.data[i]
            i=l
    }else{
        break
    }
    }
}

func topKFrequent(nums []int, k int) []int {
  if len(nums)<=1{
      return nums
  }
  m:=make(map[int]int)
  for _,n:=range nums{
      m[n]++
  }
  var data []d
  for k,v:=range m{
    data=append(data,d{num:k,cnt:v})
  }
  fmt.Println(m,data)
  if len(data)<=k{
  return getNum(data)
  }
   adjust(data,0,len(data)-1,k)
  return getNum(data[:k])
}

func adjust(data[]d,start,end,t int){
    if start <0 ||end>len(data)-1{
        return 
    }
   l:=start
   r:=end
   for l<r{ 
        for l<r{ 
        if data[l].cnt>=data[r].cnt{
           r--
        }else{
            data[l],data[r]=data[r],data[l]
            break
        }
        }
        
        for l<r{  
         if data[r].cnt<=data[l].cnt{
            l++
        }else{
             data[r],data[l]=data[l],data[r]
             break
        }
        }
  }
  if l+1==t{
     return 
  }
  if l+1<t{
     adjust(data,l+1,end,t)
     return
  }
   adjust(data,start,r-1,t)
}

func getNum(da[]d)[]int{
    var  r []int
    for _,v:=range da{
       r=append(r,v.num)
    }
    return r
}

type d struct{
    num int
    cnt int
}