leetcode 313. Super Ugly Number

最新推荐文章于 2022-08-30 15:42:26 发布

Ljosalfar

最新推荐文章于 2022-08-30 15:42:26 发布

阅读量191

点赞数

分类专栏： oj 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/usercaesar/article/details/108332235

版权

oj 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

313. Super Ugly Number

遍历
从小到大找ugly number，每个新的ugly number是primes中的素数乘积，必然是某个素数乘之前的某个ugly number。每个primes中的素数记录一个下标，对应将要乘的ugly number。每个primes中的素数，乘以之前的尽量小的ugly number，但如果之前某个ugly number能和该素数生成之前某个ugly number，则给该素数记录的接下来将要乘的ugly number下标+1。
注意INT_MAX和long long，测试数据较大。

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) 
    {
        vector<int> ugly_vec(n);
        int prime_num=primes.size();
        vector<int> prime_multip_vec(prime_num,0);
        ugly_vec[0]=1;
        if(n==1)
            return 1;
        for(int i=1;i<n;++i)
        {
//        	cout<<"i="<<i<<endl;
            int min_val=INT_MAX;
            for(int p_i=0;p_i<prime_num;++p_i)
            {
                if(prime_multip_vec[p_i]>i-1)
                    continue;
                if(1ll*primes[p_i]*ugly_vec[prime_multip_vec[p_i]]>INT_MAX)
                    continue;
                min_val=min(min_val,primes[p_i]*ugly_vec[prime_multip_vec[p_i]]);

//                printf("p_i: %d, min_val: %d\n",p_i,min_val);
            }
            ugly_vec[i]=min_val;
            for(int p_i=0;p_i<prime_num;++p_i)
            {
                if(prime_multip_vec[p_i]>i-1)
                    continue;
                if(1ll*primes[p_i]*ugly_vec[prime_multip_vec[p_i]]>INT_MAX)
                    continue;
                if(primes[p_i]*ugly_vec[prime_multip_vec[p_i]]==min_val)
                    ++prime_multip_vec[p_i];
            }
//            cout<<min_val<<endl;
        }
        return ugly_vec[n-1];
    }
};