LeetCode·面试题 17.09.第k个数·动态规划

链接：https://leetcode.cn/problems/get-kth-magic-number-lcci/solution/-by-xun-ge-v-rqxe/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。 

题目

 

思路

题目要求 求第 k 个数的值，要求为这些数的素因子至少为3,5,7之中的一个

那么我们可以枚举所有3,5，7能构成的数，并选择第k个，选择第k个的方法比较多：可以用数组，可以用哈希表，可以用堆

这里用数组，数组其实就是dp数组，记录前k个数，最后寻找第k个返回即可

代码

int getKthMagicNumber(int k) {
    int dp[k + 1];
    dp[1] = 1;//初始化
    int p3 = 1, p5 = 1, p7 = 1;//对于3,5,7
    for (int i = 2; i <= k; i++) {
        int num3 = dp[p3] * 3, num5 = dp[p5] * 5, num7 = dp[p7] * 7;//3,5,7素因子对应的合数
        dp[i] = fmin(fmin(num3, num5), num7);//每次取最小值
        if (dp[i] == num3) {//判断最小值为那一个，对应p+1供下一次用
            p3++;
        }
        if (dp[i] == num5) {
            p5++;
        }
        if (dp[i] == num7) {
            p7++;
        }
    }
    return dp[k];//取前k个
}


作者：xun-ge-v
链接：https://leetcode.cn/problems/get-kth-magic-number-lcci/solution/-by-xun-ge-v-rqxe/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。