313. Super Ugly Number

来源:互联网 发布:马扎克系统编程 编辑:程序博客网 时间:2024/05/17 06:59

这道题给出一个质数数组,求出第n个超级丑陋数。

方法和之前的丑陋数一样,下面引用了别人博客里的图片,画的很清楚。这里需要一个idx数组,来记录primes里面每一个数应该和结果集里的哪一个数相乘。

这里写图片描述

class Solution {public:    int nthSuperUglyNumber(int n, vector<int>& primes) {        vector<int> res(1, 1), idx(primes.size(), 0);        while(res.size() < n){            vector<int> tmp;            int mn = INT_MAX;            for(int i = 0; i < primes.size(); ++i){                tmp.push_back(res[idx[i]] * primes[i]);            }            for(int i = 0; i < primes.size(); ++i){                mn = min(mn, tmp[i]);            }            for(int i = 0; i < primes.size(); ++i){                if(mn == tmp[i])                    ++idx[i];            }            res.push_back(mn);        }        return res.back();    }};


原创粉丝点击