[leetcode]313题解

题目的意思是要寻找寻找第n个丑数（丑数只能是给定素数的乘积）

参照网上大牛的做法，

primes[]数组，题目给定的构成丑数的素数集合

res[]数组，用来保存丑数的，即res[i]即为第i个丑数，注意res[0]=1

index[]数组，用来保存primes[j]将要乘以的res数组的索引index[j]。

如初始化时，primes数组中元素都会和res[0]相乘，取最小数，作为下一个（比1大的最小丑数）丑数，结果是2，即2与res[0]相乘已经使用过了，下次再比较最小丑数，与2相乘的就是res[1]了。

AC代码如下：

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

下面说一下，我一开始犯的一个错误，直接贴代码

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        int len=primes.size(),pos,now,min_num;
        vector<int> res(n);
        vector<int>index(len,0);
        res[0]=1;
        for(int i=1;i<n;++i)
        {
            pos=0,min_num=INT_MAX;
            for(int j=0;j<len;++j)
            {
                now=primes[j]*res[index[j]];
                if(now<min_num)
                {
                    pos=j,min_num=now;
                }
            }
            res[i]=min_num;
            index[pos]++;
        }
        return res[n-1];
    }
};

这段代码乍一看和AC代码没有什么区别，只是代码二记录了每次循环最小值和出现最小值的索引，并把索引值+1

但是仔细研究就会发现，代码二是有问题的，如当给定素数集合是{2,3,5}的时候，再计算丑数6之后，按照代码二的处理，只会把素数2对应的index[0]（此时的index[0]为2，对应的res[2]为3）加1，而不会把素数3对应的index[1]（此时的index[1]为1，对应的res[1]为2）加1，但是3*2和2*3都是6，所以素数2和素数3对应的索引都应该加1，因此导致错误。