[leetcode] Super Ugly Number

用时: 24ms

思路：

因为super ugly number一定是以primes数列里的元素为因子得到的乘积，要找出第n个super ugly number，等价于求第n小的 以primes内元素为因子的乘积。

-> 用answer数组存放super ugly number，一直算到answer数列的第n个元素即为答案。

-> 因为answer[k]为第k小的super ugly number，answer[k]一定是由answer数组中在它之前的某个元素乘以primes数组中的某个元素得到的最小的乘积。

-> 如果primes中的某个元素primes[aa]已经乘过answer中的某个元素answer[bb]得到乘积放入answer[cc]中，则primes[aa]以后不能再乘answer[bb]，否则会出现重复。为记忆，设置大小与primes[]相同的vector t，t[i]表示primes[i]当前要用于计算乘积时要对应的super ugly number的下标，相当于用于标记primes[i]当前瞄准到answer[]中哪个元素的一个指针。即当前如果要用primes[i]乘积，乘积应为primes[i]*answer[t[i]]。每当确定一个乘积为正确值放入answer[]中后，t[i]+=1。

-> possible_current_answer为大小和primes相同的数组，存放当前primes中各元素与其对应的answer中元素相乘的乘积，当前要求的answer[k]即从中产生（选出其中最小值）。

-> 另：在一次求super ugly number的计算中，当两个不同的primes中元素(e.g. primes[a]和primes[b])乘各自对应的answer中元素得到的乘积相同时（如2*3和3*2)，则应该把它们各自瞄准的answer中元素位置都向前移动以避免未来出现重复。i.e. 将t[a]和t[b]都+1。在实现中表现为：

//计算并填入answer[k]后
for (int i=0;i<ps;i++) {
                if (answer[k]==possible_current_answer[i])
                    ++t[i];
}

实现：

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        int ps=primes.size();
        int answer[n], possible_current_answer[ps];
        vector<int> t(ps,0);
        answer[0]=1;
        for (int k=1;k<n;k++) {
            //answer[k]=max{answer[t[i]]*primes[i], i=0:ps-1.}
            for (int i=0;i<ps;i++)
                possible_current_answer[i]=answer[t[i]]*primes[i];
            answer[k]=*min_element(possible_current_answer, possible_current_answer+ps);
            for (int i=0;i<ps;i++) {
                if (answer[k]==possible_current_answer[i])
                    ++t[i];
            }
        }
        return answer[n-1];
    }
};

 

转载于:https://www.cnblogs.com/RDaneelOlivaw/p/10346989.html