Leetcode264. 丑数 II C++

题目描述

编写一个程序，找出第 n 个丑数。

丑数就是只包含质因数 2, 3, 5 的正整数。

示例:

输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:

1 是丑数。
n 不超过1690。

思路

三指针+动态规划 ,难度不小
第一个丑数是1，之后的所有丑数都可以通过它之前的丑数乘以2,3,5来得到。那么从第一个丑数1开始，之后所有的数都通过这个方法得到。唯一需要确定的就是下一个丑数是通过哪个数乘以多少来得到的。

每个数都可以乘以2, 3, 5来得到新的丑数，所以通过三个指针来表示乘2,乘3,乘5进行到的位置。
如果下一个数是通过乘以2到的，那么index2加一。其他同理。

用三个指针来实现。这里三个指针还有去重的功能，使用三个并列的 if 判断语句，当重复的时候，两个指针对应处都加1。例如，再得到1,2,3,4,5之后，index2指向3， index3指向2，index5指向2。这个时候，index2这里将得到6，index3这里也将得到6，所以两个指针同时后移。

解答

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> dp(n,1);
        int index2=0, index3=0, index5=0;
        for(int i=1;i<n;++i)
        {
            dp[i]=min(dp[index2]*2,dp[index3]*3,dp[index5]*5);
            if(dp[i]==dp[index2]*2) ++index2;
            if(dp[i]==dp[index3]*3) ++index3;
            if(dp[i]==dp[index5]*5) ++index5;
        }
        return dp[n-1];
    }
    int min(int a, int b, int c)
    {
        return a<b?(a<c?a:c):(b<c?b:c);
    }
};