剑指offer49.丑数

最新推荐文章于 2022-04-23 14:32:12 发布

ainstinct

最新推荐文章于 2022-04-23 14:32:12 发布

阅读量132

点赞数 1

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/weixin_40392620/article/details/107381115

版权

数据结构与算法专栏收录该内容

97 篇文章 0 订阅

订阅专栏

题目：我们把只包含质因子 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];
  }
}