leetcode 264. Ugly Number II 计算第K个丑数

最新推荐文章于 2025-04-12 19:07:09 发布

JackZhangNJU

最新推荐文章于 2025-04-12 19:07:09 发布

阅读量427

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/jackzhang_123/article/details/78109430

leetcode For C++ 同时被 3 个专栏收录

551 篇文章

订阅专栏

leetcode For Java

502 篇文章

订阅专栏

需要好好想一下的题目

355 篇文章

订阅专栏

本文介绍了一种高效算法来找到第N个丑数。丑数是指只包含质因数2、3和5的正整数。文中提供了Java和C++两种实现方式，并通过实例详细解释了算法的工作原理。

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

Write a program to find the n-th ugly number.

Ugly numbers are positive numbers whose prime factors only include 2, 3, 5. For example, 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 is the sequence of the first 10 ugly numbers.

Note that 1 is typically treated as an ugly number, and n does not exceed 1690.

题意很简单，上一道题是除法，这一道题是乘法。

建议和上一道题leetcode 263. Ugly Number和 leetcode 313. Super Ugly Number 超级丑数一起学习。

代码如下：

import java.util.SortedSet;
import java.util.TreeSet;

/*
 * 这个很棒，需要学习
 * 
 * */
class Solution {
    public int nthUglyNumber(int n) 
    {
        SortedSet<Long> set=new TreeSet<>();
        set.add((long)1);

        long one=set.first();
        for(int i=0;i<n;i++)
        {
            one=set.first();
            set.add(one*2);
            set.add(one*3);
            set.add(one*5);
            set.remove(one);
        }
        return (int)(one);
    }

}

下面是C++的做法，上一道题是除法做法，这一道题是乘法做法，设置三个指针即可，代码如下：

#include <iostream>
#include <vector>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <queue>
#include <stack>
#include <string>
#include <climits>
#include <algorithm>
#include <sstream>
#include <functional>
#include <bitset>
#include <numeric>
#include <cmath>
#include <regex>

using namespace std;



class Solution
{
public:
    int nthUglyNumber(int n)
    {
        vector<long long> res;
        set<long long> ss;
        ss.insert(1);
        while (res.size() < n)
        {
            long long a = *(ss.begin());
            ss.insert(a * 2);
            ss.insert(a * 3);
            ss.insert(a * 5);
            ss.erase(a);
            res.push_back(a);
        }
        return res[n - 1];
    }

    int nthUglyNumberByCala(int n) 
    {
        vector<int> res(1,1);
        int i = 0, j = 0, k = 0;
        while (res.size() < n)
        {
            int a = min(res[i] * 2, min(res[j] * 3, res[k] * 5));
            res.push_back(a);
            if (a == res[i] * 2)
                i++;
            if (a == res[j] * 3)
                j++;
            if (a == res[k] * 5)
                k++;
        }
        return res[n - 1];
    }
};