面试题 17.09. 第 k 个数

2022-09-28 09:11:07 920 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

武汉理工大学 python实验三请编一个程序，用户在同一行内输入两个整数，代表头和脚的数量，编程计算笼中各有多少只鸡和兔（假设鸡和兔都正常，无残疾），如无解则输出`“Data Error!题目为：今有物不知其数，三三数之剩二，五五数之剩三，七七数之剩二，问物几何？年前，《孙子算经》中记载一个有趣的问题：今有雉兔同笼，上有三十五头，下有九十四足，问雉兔各几何？有一座八层宝塔，每一层都有一些琉璃灯，每一层的灯数都是上一层的二倍，已知共有。编写一个宝塔上的琉璃灯的小程序，计算并输出每层各有多少盏琉璃灯。，如果有多个解，则分行输出，每行一个解；阅读详情

题意

有些数的素因子只有 3，5，7，请设计一个算法找出第 k 个数。注意，不是必须有这些素因子，而是必须不包含其他的素因子。例如，前几个数按顺序应该是 1，3，5，7，9，15，21。

示例 1:
输入: k = 5
输出: 9
注意大数问题

题解

class Solution {
    public int getKthMagicNumber(int k) {
        Set<Long> set = new HashSet<Long>();
        PriorityQueue<Long> queue = new PriorityQueue<Long>();
        queue.offer(1L);
        long t=0;
        while(!queue.isEmpty()&&k>0){
            t = queue.poll();
            k--;
            if(!set.contains(t*3)) {queue.offer(t*3);set.add(t*3);}
            if(!set.contains(t*5)) {queue.offer(t*5);set.add(t*5);}
            if(!set.contains(t*7)) {queue.offer(t*7);set.add(t*7);}
        }
        return (int)t;
    }
}