poj - 1338 - Ugly Numbers(优先队列)

本文介绍了一种使用小根堆实现的算法,用于找出第n小的丑数(只包含质因数2、3、5)。通过优先队列进行筛选,最终输出指定位置的丑数。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题意:问第n(n <= 1500)小的丑数(质因数只有2或3或5)是几。

题目链接:http://poj.org/problem?id=1338

——>>1, 2, 3, 4, 5, 6, 8, ...

假设小根堆存以上丑数,那么每次取出最小的数,这个最小的数nMin,它可以生成三个数:nMin * 2, nMin * 3, nMin * 5,将这三个数放入小根堆继续,一直复筛出1500个丑数为止。

小根堆可用优先队列来替代。

#include <cstdio>
#include <queue>
#include <vector>

using std::priority_queue;
using std::vector;
using std::greater;

const int MAXN = 1500;

long long nRet[MAXN];

void GetUglyNumbers()
{
    int nCnt = 0;
    priority_queue<long long, vector<long long>, greater<long long> > pq;
    pq.push(1);

    while (nCnt < MAXN)
    {
        long long nMin = pq.top();
        pq.pop();
        if (nRet[nCnt] != nMin)
        {
            nRet[++nCnt] = nMin;
            pq.push(nMin * 2);
            pq.push(nMin * 3);
            pq.push(nMin * 5);
        }
    }
}

int main()
{
    int n;

    GetUglyNumbers();
    while (scanf("%d", &n) == 1 && n)
    {
        printf("%I64d\n", nRet[n]);
    }

    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值