Fork me on GitHub

版权声明 本站原创文章 由 萌叔 发表
转载请注明 萌叔 | https://vearne.cc

题目

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/super-ugly-number

编写一段程序来查找第 n 个超级丑数。超级丑数是指其所有质因数都是长度为 k 的质数列表 primes 中的正整数。

示例

输入: n = 12, primes = [2,7,13,19]
输出: 32 
解释: 给定长度为 4 的质数列表 primes = [2,7,13,19],前 12 个超级丑数序列为:[1,2,4,7,8,13,14,16,19,26,28,32]。
说明:

1 是任何给定 primes 的超级丑数。
 给定 primes 中的数字以升序排列。
0 < k ≤ 100, 0 < n ≤ 106, 0 < primes[i] < 1000 。
第 n 个超级丑数确保在 32 位有符整数范围内。

解题思路1

由于这是一道中等难度的题目,萌叔第一想到的是穷举,从1开始穷举,1、2 3、... N 判断每个数是否是超级丑数。这个思路肯定是没错的,结果提交后发现超时了。超时的原因是扫描了太多无用的数据。

解题思路2

超级丑数是指其所有质因数都在primes 列表中, 那么很容易可以想到,可以用primes列表中的质数来构造超级丑数。现在唯一麻烦的是怎么按照超级丑数的大小来定序。
查找第 n 个超级丑数,也就是要从整个解空间中,找出第N大的超级丑数。
我们先来看看解空间有什么特点。
为了简化图表,假定primes = [2,7,13,19],1是比较特殊的情况,跳过不看

解空间1

  • 解空间是一颗多叉数
  • 从根沿着树干节点向下,节点的数值越来越大
  • 每个节点的孩子节点,都是它的值乘以primes 中的每个数
  • 当前的最小值,可能在多个子树中产生

某一时刻,假定我们已经找到了超级丑数12,我们把12、从树中移除,可以得到若干个新的子树,新的子树的根节点分别是
2*22*72*1378,下一个最小的超级丑数,只能从这几个数产生。因为剩下的节点都是以上节点的子孙节点,所以肯定要比这些节点中最小值还要大。

思路确定

每次从解空间中找到一个超级丑数,就从把它从多叉树上移除,然后扫描剩余多个子树的根节点,确定下一个超级丑数。为了加快扫描的速度,已知的子树的根,都存储在一个小顶堆里。

递推公式

f(n) = min(f(n -1)的子节点, 其它子树的根)

代码如下

package main

import (
    "container/heap"
    "fmt"
)

// An IntHeap is a min-heap of ints.
type IntHeap []int

func (h IntHeap) Len() int           { return len(h) }
func (h IntHeap) Less(i, j int) bool { return h[i] < h[j] }
func (h IntHeap) Swap(i, j int)      { h[i], h[j] = h[j], h[i] }

func (h *IntHeap) Push(x interface{}) {
    // Push and Pop use pointer receivers because they modify the slice's length,
    // not just its contents.
    *h = append(*h, x.(int))
}

func (h *IntHeap) Pop() interface{} {
    old := *h
    n := len(old)
    x := old[n-1]
    *h = old[0 : n-1]
    return x
}

func nthSuperUglyNumber(n int, primes []int) int {
    h := IntHeap([]int{})
    heap.Push(&h, 1)
    for _, value := range primes {
        heap.Push(&h, value)
    }
    idx := 0
    last := 0
    var t int
    for len(h) > 0 && idx < n {
        t = heap.Pop(&h).(int)
        //  解决冗余数据
        for t <= last {
            t = heap.Pop(&h).(int)
        }
        last = t
        for _, v := range primes {
            heap.Push(&h, v*last)
        }
        idx++
    }
    return last
}

func main() {
    n := 550
    primes := []int{3, 7, 17, 19, 23, 29, 31, 37, 43,
        47, 59, 67, 71, 73, 83, 97, 103, 109, 127, 131, 139, 149, 163, 167, 173, 191, 197, 233, 241, 251}
    fmt.Println(nthSuperUglyNumber(n, primes))
}

请我喝瓶饮料

微信支付码

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注