【Leetcode】264. Ugly Number II

题目地址：

https://leetcode.com/problems/ugly-number-ii/

一个正整数是丑数，含义是它的素因子只有$2,3,5$。规定$1$是第$1$个丑数，问第$n$个丑数是几。

思路是多路归并。设丑数集合为$S$，考虑三个序列：$$\begin{gathered} S_2=[2\times 1,2\times 2,2\times 3,...]=2S \\ {S}_3=[3\times 1,3\times 2,3\times 3,...]=3S \\ {S}_5=[5\times 1,5\times 2,5\times 3,...]=5S \end{gathered}$$也就是说，$S_i$就是所有含素因子$i$的丑数。求第$n$个丑数是几，相当于在求这三个序列里第$n-1$小的数是几。用三个指针计数即可。由于$S_i$构造的特殊性，我们可以直接将三个指针直接指向我们要构造的序列$S$。第一个丑数是$1$，可以先令$S[0]=1$，想象有三个指针$i,j,k$都指向$S[0]$，然后每次看一下$2\times S[i],3\times S[j],5\times S[k]$哪个最小，然后append到$S$后面，将那个最小的数对应的指针后移。需要注意的是，如果某个数丑数被枚举了多次，需要将对应的指针同时后移。例如，容易知道前五项$S=[1,2,3,4,5,...]$，接下来要枚举的丑数是$6$，但是$6=2\times 3=3\times 2$，会同时被$i$和$j$枚举到，枚举一次之后要将两个指针同时后移，避免重复枚举。代码如下：

class Solution {
 public:
  int nthUglyNumber(int n) {
    vector<int> a(n);
    a[0] = 1;
    int idx = 1;
    for (int two = 0, three = 0, five = 0; idx < n;) {
      int ne = min(min(a[two] * 2, a[three] * 3), a[five] * 5);
      a[idx++] = ne;
      if (ne == a[two] * 2) two++;
      if (ne == a[three] * 3) three++;
      if (ne == a[five] * 5) five++;
    }

    return a.back();
  }
};

时空复杂度$O(n)$。