获取第M个水仙花数

最新推荐文章于 2023-11-04 19:16:43 发布

south_sheng

最新推荐文章于 2023-11-04 19:16:43 发布

阅读量87

点赞数

分类专栏：算法 python 文章标签：算法数据结构

本文链接：https://blog.csdn.net/south_sheng/article/details/126572590

版权

算法同时被 2 个专栏收录

18 篇文章 1 订阅

订阅专栏

python

15 篇文章 1 订阅

订阅专栏

"""
/*
所谓的水仙花数是指一个n位的正整数其各位数字的n次方的和等于该数本身，
例如153=1^3+5^3+3^3,153是一个三位数
输入描述
    第一行输入一个整数N，
    表示N位的正整数N在3-7之间包含3,7
    第二行输入一个正整数M，
    表示需要返回第M个水仙花数
输出描述
    返回长度是N的第M个水仙花数，
    个数从0开始编号，
    若M大于水仙花数的个数返回最后一个水仙花数和M的乘积，
    若输入不合法返回-1

示例一：

    输入
     3
     0
    输出
     153
    说明：153是第一个水仙花数
 示例二：
    输入
    9
    1
    输出
    -1
 */
"""

 import math
    N = int(input().strip())
    M = int(input().strip())
    MinNum = int(math.pow(10,N-1))
    MaxNum = int(math.pow(10,N))
    res = list()
    if N < 3 or N > 7:
        print(-1)
    else:
        for i in range(MinNum,MaxNum):
            summ = 0
            bit = MinNum
            while bit != 1:
                print(i//bit)
                summ += int(math.pow(i//bit%10,N))
                bit //= 10
            summ += int(math.pow(i%10,N))
            if summ == i:
                res.append(summ)

            if len(res)-1 == M:
                print(res[M])
                break
            
        if M > len(res):
            print(res[-1]*M)