数据结构与算法(31):丑数

剑指 Offer 49. 丑数

我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。

题解(三指针)

这里我也没什么思路,看评论大佬的,以下都是复制过来的

我们知道,丑数的排列肯定是1,2,3,4,5,6,8,10… 然后有一个特点是,任意一个丑数都是由小于它的某一个丑数2,3或者5得到的,那么如何得到所有丑数呢? 现在假设有3个数组,分别是: A:{12,22,32,42,52,62,82,10*2…}

B:{13,23,33,43,53,63,83,103…}

C:{15,25,35,45,55,65,85,105…}

那么所有丑数的排列,必定就是上面ABC3个数组的合并结果然后去重得到的,那么这不就转换成了三个有序数组的无重复元素合并的问题了吗?而这三个数组就刚好是{1,2,3,4,5,6,8,10…}乘以2,3,5得到的。

合并有序数组的一个比较好的方法,就是每个数组都对应一个指针,然后比较这些指针所指的数中哪个最小,就将这个数放到结果数组中,然后该指针向后挪一位。

回到本题,要求丑数ugly数组中的第n项,而目前只知道ugly[0]=1,所以此时三个有序链表分别就只有一个元素:

A : {1*2…}

B : {1*3…}

C :{1*5…}

假设三个数组的指针分别是i,j,k,此时均是指向第一个元素,然后比较A[i],B[j]和C[k],得到的最小的数A[i],就是ugly[1],此时ugly就变成{1,2}了,对应的ABC数组就分别变成了:

A : {12,22…}

B : {13, 23…}

C :{15,25…}

此时根据合并有序数组的原理,A数组指针i就指向了下一个元素,即’2*2’,而j和k依然分别指向B[0]和C[0],然后进行下一轮合并,就是A[1]和B[0]和C[0]比较,最小值作为ugly[2]…如此循环n次,就可以得到ugly[n]了。

此外,注意到ABC三个数组实际上就是ugly[]*2,ugly[]*3和ugly[]*5的结果,所以每次只需要比较A[i]=ugly[i]*2,B[j]=ugly[j]*3和C[k]=ugly[k]*5的大小即可。然后谁最小,就把对应的指针往后移动一个,为了去重,如果多个元素都是最小,那么这多个指针都要往后移动一个。

class Solution {
    public int nthUglyNumber(int n) {
        if(n==0)return 0;
        int[] ugly=new int[n];
        ugly[0] = 1;     //基础丑数为1
        int i=0,j=0,k=0;  //初始分别指向三个有序链表第一个元素,这三个有序链表是想象出来的,分别就是ugly数组元素分别乘以2,3,5得到的
        for(int idx=1;idx<n;idx++)
        {
            int tmp =Math.min(ugly[i]*2,Math.min(ugly[j]*3,ugly[k]*5));    
            //三个链表可能有相同元素,所以只要是最小的,都要移动指针
            if(tmp == ugly[i]*2)i++;
            if(tmp == ugly[j]*3)j++;
            if(tmp == ugly[k]*5)k++;
            ugly[idx] = tmp;
        }
        return ugly[n-1];
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值