OJ笔记 19116 丑数

19116 丑数

Description
“丑数”是指除了质因子2,3，5，不含其它质因子的正整数，例如由小到大前10个“丑数”为1, 2, 3, 4, 5, 6, 8, 9, 10, 12, …现要求编写一个程序，输出指定第几位的“丑数”。

输入格式
第一行为正整数T（T<=10000）, 表示case的数目。
此后T行，每行一个正整数 n (n <= 100000000).

输出格式
每一个n，输出第n个“丑数”

输入样例
3
1
2
9

输出样例
1
2
10

并不用每次都尝试所有组合，我们需要过滤掉不可能的组合 已知2，3，5与已有丑数相乘时，数值增长速度是不一样的
我们每次从三组新丑数中，挑选大于已知末尾丑数的最小值，来作为新的丑数，例如放入乘2的，乘3或者乘5的新丑数，如果某次未被选中放入数轴，那么肯定有某一刻会被放进去
所以使用三个游标来确定已被放入数轴的新丑数最小值，每次仅需要对比这三个游标与2，3，5乘积即可

#include <iostream>
#include <cstdio>
#include <string.h>
#include <math.h>
#include <vector>

using namespace std;


//不判断是否为丑数，而是从第一个丑数开始递推丑数（*2、*3、*5）
//定义丑数函数
int Chou(int x)
{
    if(x < 7){
        return x;//小于7的正整数都是丑数,返回本身
    }
    int index_2 = 0,index_3 = 0,index_5 = 0;//游标，已放入数组的新丑数最小值下标
    int t2 = 0,t3 = 0,t5 = 0;//每次出现的丑数
    vector<int> dp(x);//动规数组
    dp[0] = 1;//初始元素为1，第一个丑数为1
    for(int i = 1;i < x;i++){
        //每次都分别* 2、3、5，挑选最小的放入丑数数组中
        t2 = dp[index_2] * 2;//首个 * 2 后 > dp[i-1]（上一个丑数）的丑数
        t3 = dp[index_3] * 3; //首个 * 3 后 > dp[i-1]（上一个丑数）的丑数
        t5 = dp[index_5] * 5; //首个 * 5 后 > dp[i-1]（上一个丑数）的丑数
        dp[i] = min(t2,min(t3,t5));
        //相应的游标往下移
        if(dp[i] == t2) index_2++;
        if(dp[i] == t3) index_3++;
        if(dp[i] == t5) index_5++;
    }
    return dp[x - 1];//返回上一个丑数（即第n个丑数，因为dp数组从0开始的）
}


main()
{
	int T;
	scanf("%d",&T);

	for(int i = 1;i <= T;i++){
        int n;
        scanf("%d",&n);
        int ans = Chou(n);
        cout << ans << endl;
	}

  	return 0;
}

参考：
[剑指offer]面试题第[49]题[Leetcode][第264题][JAVA][丑数][动态规划][堆]
leetcode第264题 丑数|| （动态规划）