质数生成集

质数生成集

求第$k$个丑数。

丑数$N$的质因数只有$2,3,5$。

我们知道任何一个丑数$N$，都是小于$N$的某一个丑数乘以$2,3,5$生成的。因此，我们得到三个有序序列：

$$\begin{gathered} 1\ast 2,2\ast 2,3\ast 2,4\ast 2,5\ast 2,\dots \\ 1\ast 3,2\ast 3,3\ast 3,4\ast 3,5\ast 3,\dots \\ 1\ast 5,2\ast 5,3\ast 5,4\ast 5,5\ast 5,\dots \end{gathered}$$

本问题就相当于从一个合并三个虚构的有序序列即可。

设置三个指针$p2,p3,p5$分别指向各自序列的下标，依次取数即可。

UVA136

#include <bits/stdc++.h>

using namespace std;

#define FR freopen("in.txt", "r", stdin)
#define FW freopen("out.txt", "w", stdout)

#define MAXT INT_MAX

typedef long long ll;

int dp[1600];

int main()
{
    dp[1] = 1;
    int p2 = 1;
    int p3 = 1;
    int p5 = 1;

    for(int i = 2;i <= 1500;i++)
    {
        int val = min({dp[p2] * 2,dp[p3] * 3,dp[p5] * 5});

        dp[i] = val;

        if(val == dp[p2] * 2) p2++;
        if(val == dp[p3] * 3) p3++;
        if(val == dp[p5] * 5) p5++;
    }

    printf("The 1500'th ugly number is %d.",dp[1500]);
    return 0;
}

此方法可以推广至任意素因子。