剑指offer49.丑数

题目:我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
本题链接:https://leetcode-cn.com/problems/chou-shu-lcof/solution/chou-shu-ii-qing-xi-de-tui-dao-si-lu-by-mrsate/
在这里插入图片描述
在这里插入图片描述

class Solution {
    public int nthUglyNumber(int n) {
        int[] dp = new int[n];
        dp[0] = 1;//初始条件
        int a =0,b=0,c=0;
        //
        for(int i=1;i<n;i++){
            int n1 = dp[a]*2;
            int n2 = dp[b]*3;
            int n3 = dp[c]*5;
            dp[i] = Math.min(Math.min(n1,n2),n3);//倍数最小的就是,因为题目要求是从小到大排列
            //更新a b c的指针,并去除重复的
            if(dp[i] == n1) a++;
            if(dp[i] == n2) b++;
            if(dp[i] == n3) c++;
        }
        return dp[n-1];
    }
}

在这里插入图片描述
解法二:使用堆 PriorityQueue(优先队列,默认从小到大)【add,offer,peek,poll,remove】
我们从堆中包含一个数字开始:1,去计算下一个丑数。将 1 从堆中弹出然后将三个数字添加到堆中:1×2, 1×3和 1×5。现在堆中最小的数字是 2。为了计算下一个丑数,要将 2 从堆中弹出然后添加三个数字:2×2, 2×3和2×5。重复该步骤计算所有丑数。在每个步骤中,弹出堆中最小的丑数 kk,并在堆中添加三个丑数:k×2, k×3,和k×5。
为了控制重复的数,使用一个map用来记录数,在添加数据时,如果有重复的则不放入堆中

class Ugly {
  public int[] nums = new int[1690];
  Ugly() {
    HashSet<Long> seen = new HashSet();
    PriorityQueue<Long> heap = new PriorityQueue<Long>();
    seen.add(1L);
    heap.add(1L);
    long currUgly, newUgly;
    int[] primes = new int[]{2, 3, 5};
    for(int i = 0; i < 1690; ++i) {
      currUgly = heap.poll();
      nums[i] = (int)currUgly;
      for(int j : primes) {
        newUgly = currUgly * j;
        if (!seen.contains(newUgly)) {
          seen.add(newUgly);
          heap.add(newUgly);
        }
      }
    }
  }
}

class Solution {
  public static Ugly u = new Ugly();
  public int nthUglyNumber(int n) {
    return u.nums[n - 1];
  }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值