Q:

Leetocde的两道丑数题目:264. 丑数 II➕313. 超级丑数-LMLPHP

A:

用变量记录已经✖2、✖3、✖5的元素下标i2、i3、i5。表示截止到i2的元素都已经乘过2(结果添加到序列尾部的意思),i3、i5同理。这样每次可以循环可以O(1)时间找到下一个最小的丑数,时间O(N),空间O(N)。

class Solution {
public:
int nthUglyNumber(int n) {
if(n<=0){
return 0;
}
int i2=0,i3=0,i5=0;
vector<int> vec={1};
while(vec.size()<n){
int min_mul=vec[i2]*2;
min_mul=min(min_mul,vec[i3]*3);
min_mul=min(min_mul,vec[i5]*5);
if(min_mul==vec[i2]*2){
i2++;
}
if(min_mul==vec[i3]*3){
i3++;
}
if(min_mul==vec[i5]*5){
i5++;
}
vec.push_back(min_mul);
}
return vec.back();
}
};

还有一种利用堆的方法,堆初始只有一个元素1,第一次循环:堆顶元素1分别✖2、3、5,结果加入堆,堆顶元素1就是第一个丑数,pop掉。这样堆还剩2、3、5。第二次循环:堆顶元素2分别✖2、3、5,结果加入堆,堆顶元素2就是第二个丑数。

这样循环n次,最终会找到第n个丑数。时间O(NlogN),空间O(N)。

Q:

Leetocde的两道丑数题目:264. 丑数 II➕313. 超级丑数-LMLPHP

A:

之前是2、3、5,现在是给定了primes.size()个数。那么就建一个数组pos,其中pos[i]指明结果序列中已经乘了primes[i]的最小下标,和上一题类似。

class Solution {
public:
int nthSuperUglyNumber(int n, vector<int>& primes) {
if(n<=0){
return 0;
}
vector<int> vec={1};
vector<int> pos(primes.size(),0);
while(n>1){
--n;
int nex=INT32_MAX;
for(int i=0;i<primes.size();++i){
nex=min(nex,vec[pos[i]]*primes[i]);
}
for(int i=0;i<primes.size();++i){
if(nex==vec[pos[i]]*primes[i]){
pos[i]++;
}
}
vec.push_back(nex);
}
return vec.back();
}
};
05-16 02:18