leetcode 313 : Super Ugly Number

原创于 2015-12-07 12:03:11 发布 · 761 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #面试 #leetcode

编程语言学习专栏收录该内容

34 篇文章

订阅专栏

本文探讨了如何通过编程找到特定条件下的超优数。这些超优数是正整数，其所有质因数都来自一组给定的质数列表。文章详细介绍了算法的实现过程，包括初始化、迭代求解以及最终结果的获取。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、原题如下：

Write a program to find the nth super ugly number.

Super ugly numbers are positive numbers whose all prime factors are in the given prime list primes of size k. For example, [1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32] is the sequence of the first 12 super ugly numbers given primes = [2, 7, 13, 19] of size 4.

Note:
(1) 1 is a super ugly number for any given primes.
(2) The given numbers in primes are in ascending order.
(3) 0 < k ≤ 100, 0 < n ≤ 106, 0 < primes[i] < 1000.

2、解题如下：

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        int size=primes.size();
        vector<int> count(size, 0), ugly_num(n, INT_MAX);
        ugly_num[0]=1;
        for(int i=1; i<n; i++){
             for(int j=0; j<primes.size(); j++){
                 ugly_num[i]=min(ugly_num[i],ugly_num[count[j]]*primes[j]);
             }
             for(int j=0; j<primes.size(); j++){
                 count[j]+=(ugly_num[i]==ugly_num[count[j]]*primes[j]);
             }
        }
        return ugly_num[n-1];
    }
};